Package jdistlib.accelerator

Interface PreparedDenseMatrix

All Superinterfaces:
AutoCloseable
public interface PreparedDenseMatrix extends AutoCloseable
Reusable FP64 dense matrix handle; providers may retain storage on device.

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    close()
     
    int
    columns()
     
    void
    multiply(MatrixTranspose transpose, double alpha, double[] right, int rightColumns, double beta, double[] result)
    Performs C := alpha*op(A)*B + beta*C with row-major B and C.
    default void
    multiplyBatched(MatrixTranspose transpose, double alpha, double[][] right, int rightColumns, double beta, double[][] result)
     
    int
    rows()
     

  • Method Details

    • rows

      int rows()

    • columns

      int columns()

    • multiply

      void multiply(MatrixTranspose transpose, double alpha, double[] right, int rightColumns, double beta, double[] result)
      Performs C := alpha*op(A)*B + beta*C with row-major B and C.

    • multiplyBatched

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

    • close

      void close()
      Specified by:
      close in interface AutoCloseable