Package jdistlib.accelerator

Interface PreparedFloatDenseMatrix

All Superinterfaces:

AutoCloseable

public interface PreparedFloatDenseMatrix
extends AutoCloseable

Reusable FP32 dense matrix handle; providers may retain storage on device.

Method Summary

Modifier and Type	Method	Description
void	close()
int	columns()
void	multiply(MatrixTranspose transpose, float alpha, float[] right, int rightColumns, float beta, float[] result)
default void	multiplyBatched(MatrixTranspose transpose, float alpha, float[][] right, int rightColumns, float beta, float[][] result)
int	rows()

Method Details

rows

int rows()

columns

int columns()

multiply

void multiply(MatrixTranspose transpose,
 float alpha,
 float[] right,
 int rightColumns,
 float beta,
 float[] result)

multiplyBatched

default void multiplyBatched(MatrixTranspose transpose,
 float alpha,
 float[][] right,
 int rightColumns,
 float beta,
 float[][] result)

close

void close()

Specified by:

close in interface AutoCloseable