Function std::rocm_gemm

#include <std/ml/gpu_rocm.h> source · definition

int rocm_gemm(
    struct RocmBuf a,
    int ta,
    struct RocmBuf b,
    int tb,
    struct RocmBuf c,
    unsigned long m,
    unsigned long n,
    unsigned long k,
)

c[m,n] = op(a)[m,k] * op(b)[k,n] (rocBLAS sgemm)