std :: ml::cuda_ffi
Function std::cublasLtMatmul
int cublasLtMatmul(
CublasLtHandle_* handle,
CublasLtDesc_* desc,
const float* alpha,
const CuDevPtr_* A,
CublasLtLayout_* Adesc,
const CuDevPtr_* B,
CublasLtLayout_* Bdesc,
const float* beta,
const CuDevPtr_* C,
CublasLtLayout_* Cdesc,
CuDevPtr_* D,
CublasLtLayout_* Ddesc,
const unsigned long* algo,
CuDevPtr_* workspace,
size_t wsBytes,
CudaNativeStream* stream,
)algo: a heuristic result's leading cublasLtMatmulAlgo_t