Function gemm_ex 

pub unsafe fn gemm_ex(
    handle: cublasHandle_t,
    transa: cublasOperation_t,
    transb: cublasOperation_t,
    m: c_int,
    n: c_int,
    k: c_int,
    alpha: *const c_void,
    a: CUdeviceptr,
    a_type: cudaDataType,
    lda: c_int,
    b: CUdeviceptr,
    b_type: cudaDataType,
    ldb: c_int,
    beta: *const c_void,
    c: CUdeviceptr,
    c_type: cudaDataType,
    ldc: c_int,
    compute_type: cublasComputeType_t,
    algo: cublasGemmAlgo_t,
) -> Result<(), GpuError>

cublasGemmEx — typed-erased gemm with a separate compute type.

Safety

a/b/c must point to device buffers with the dtypes encoded in a_type/b_type/c_type and the sizes implied by (m,n,k,lda,ldb,ldc).