The nvmath.bindings module is experimental and potentially subject to future changes.

gemm_ex

nvmath.bindings.cublas.gemm_ex(

intptr_t handle,

int transa,

int transb,

int m,

int n,

int k,

intptr_t alpha,

intptr_t a,

int atype,

int lda,

intptr_t b,

int btype,

int ldb,

intptr_t beta,

intptr_t c,

int ctype,

int ldc,

int compute_type,

int algo,

)

See cublasGemmEx.