gemm#

nvmath.bindings.cublasMp.gemm(
intptr_t handle,
int trans_a,
int trans_b,
int64_t m,
int64_t n,
int64_t k,
intptr_t alpha,
intptr_t a,
int64_t ia,
int64_t ja,
intptr_t desc_a,
intptr_t b,
int64_t ib,
int64_t jb,
intptr_t desc_b,
intptr_t beta,
intptr_t c,
int64_t ic,
int64_t jc,
intptr_t desc_c,
int compute_type,
intptr_t d_work,
size_t workspace_size_in_bytes_on_device,
intptr_t h_work,
size_t workspace_size_in_bytes_on_host,
)[source]#

See cublasMpGemm.