matmul_buffer_size#

nvmath.bindings.cublasMp.matmul_buffer_size(
intptr_t handle,
intptr_t matmul_desc,
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,
intptr_t d,
int64_t id,
int64_t jd,
intptr_t desc_d,
) tuple[source]#

See cublasMpMatmul_bufferSize.