Lines Matching defs:mb
705 mgpuSpMMBufferSize(int32_t ma, int32_t mb, void *a, void *b, void *c,
709 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
722 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuSpMM(int32_t ma, int32_t mb,
728 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
740 mgpuSDDMMBufferSize(int32_t ma, int32_t mb, void *a, void *b, void *c,
744 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
757 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuSDDMM(int32_t ma, int32_t mb,
763 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
788 void *s, int32_t ma, int32_t mb, void *a, void *b, void *c, int32_t ctp,
792 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
806 mgpuSpGEMMCompute(void *s, int32_t ma, int32_t mb, void *a, void *b, void *c,
810 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
824 mgpuSpGEMMCopy(void *s, int32_t ma, int32_t mb, void *a, void *b, void *c,
828 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);
946 mgpuCuSparseLtSpMMBufferSize(void *bs, int32_t ma, int32_t mb, void *a, void *b,
961 cusparseOperation_t modeB = static_cast<cusparseOperation_t>(mb);