Gemm

General matrix-matrix multiplication: updates \(C := \alpha A^\# B^\sharp + \beta C\), where \(M^\#\) and \(M^\sharp\) are individually defined to be one of \(\{M,M^T,M^H\}\).

Note

For the best performance, A, B, and C should all be in [MC,MR] distributions.

C++ API

type GemmAlgorithm

An enum which can take the form:

• GEMM_DEFAULT

• GEMM_SUMMA_A

• GEMM_SUMMA_B

• GEMM_SUMMA_C

• GEMM_SUMMA_DOT

• GEMM_CANNON

void Gemm(Orientation orientationOfA, Orientation orientationOfB, T alpha, const Matrix<T> &A, const Matrix<T> &B, T beta, Matrix<T> &C)

void Gemm(Orientation orientationOfA, Orientation orientationOfB, T alpha, const AbstractDistMatrix<T> &A, const AbstractDistMatrix<T> &B, T beta, AbstractDistMatrix<T> &C, GemmAlgorithm alg = GEMM_DEFAULT)

C API

ElError ElGemm_i(ElOrientation orientationOfA, ElOrientation orientationOfB, ElInt alpha, ElConstMatrix_i A, ElConstMatrix_i B, ElInt beta, ElMatrix_i C)

ElError ElGemm_s(ElOrientation orientationOfA, ElOrientation orientationOfB, float alpha, ElConstMatrix_s A, ElConstMatrix_s B, float beta, ElMatrix_s C)

ElError ElGemm_d(ElOrientation orientationOfA, ElOrientation orientationOfB, double alpha, ElConstMatrix_d A, ElConstMatrix_d B, double beta, ElMatrix_d C)

ElError ElGemm_c(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_float alpha, ElConstMatrix_c A, ElConstMatrix_c B, complex_float beta, ElMatrix_c C)

ElError ElGemm_z(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_double alpha, ElConstMatrix_z A, ElConstMatrix_z B, complex_double beta, ElMatrix_z C)

ElError ElGemmDist_i(ElOrientation orientationOfA, ElOrientation orientationOfB, ElInt alpha, ElConstDistMatrix_i A, ElConstDistMatrix_i B, ElInt beta, ElDistMatrix_i C)

ElError ElGemmDist_s(ElOrientation orientationOfA, ElOrientation orientationOfB, float alpha, ElConstDistMatrix_s A, ElConstDistMatrix_s B, float beta, ElDistMatrix_s C)

ElError ElGemmDist_d(ElOrientation orientationOfA, ElOrientation orientationOfB, double alpha, ElConstDistMatrix_d A, ElConstDistMatrix_d B, double beta, ElDistMatrix_d C)

ElError ElGemmDist_c(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_float alpha, ElConstDistMatrix_c A, ElConstDistMatrix_c B, complex_float beta, ElDistMatrix_c C)

ElError ElGemmDist_z(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_double alpha, ElConstDistMatrix_z A, ElConstDistMatrix_z B, complex_double beta, ElDistMatrix_z C)

ElGemmAlgorithm

An enum which can take the form:

• EL_GEMM_DEFAULT

• EL_GEMM_SUMMA_A

• EL_GEMM_SUMMA_B

• EL_GEMM_SUMMA_C

• EL_GEMM_SUMMA_DOT

• EL_GEMM_CANNON

ElError ElGemmXDist_i(ElOrientation orientationOfA, ElOrientation orientationOfB, ElInt alpha, ElConstDistMatrix_i A, ElConstDistMatrix_i B, ElInt beta, ElDistMatrix_i C, ElGemmAlgorithm alg)

ElError ElGemmXDist_s(ElOrientation orientationOfA, ElOrientation orientationOfB, float alpha, ElConstDistMatrix_s A, ElConstDistMatrix_s B, float beta, ElDistMatrix_s C, ElGemmAlgorithm alg)

ElError ElGemmXDist_d(ElOrientation orientationOfA, ElOrientation orientationOfB, double alpha, ElConstDistMatrix_d A, ElConstDistMatrix_d B, double beta, ElDistMatrix_d C, ElGemmAlgorithm alg)

ElError ElGemmXDist_c(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_float alpha, ElConstDistMatrix_c A, ElConstDistMatrix_c B, complex_float beta, ElDistMatrix_c C, ElGemmAlgorithm alg)

ElError ElGemmXDist_z(ElOrientation orientationOfA, ElOrientation orientationOfB, complex_double alpha, ElConstDistMatrix_z A, ElConstDistMatrix_z B, complex_double beta, ElDistMatrix_z C, ElGemmAlgorithm alg)

Python API

GemmAlgorithm

An enum which can take the values…**TODO**

Gemm(orientA, orientB, alpha, A, B, beta, C, alg=GEMM_DEFAULT)