std1.0.0-beta-7 SafeC docs

std :: ml::cuda_ffi

Function std::cublasGemmEx

#include <std/ml/cuda_ffi.h> source

int cublasGemmEx(
    CublasHandle_* handle,
    int transa,
    int transb,
    int m,
    int n,
    int k,
    const float* alpha,
    const CuDevPtr_* a,
    int aType,
    int lda,
    const CuDevPtr_* b,
    int bType,
    int ldb,
    const float* beta,
    CuDevPtr_* c,
    int cType,
    int ldc,
    int computeType,
    int algo,
)

alpha / beta: float (computeType CUBLAS_COMPUTE_32F*)