rocblas库中的SGEMM方法

在rocBLAS库中，SGEMM函数用于实现单精度浮点数矩阵乘法，函数原型如下：

rocblas_status rocblas_sgemm(
    rocblas_handle handle,
    rocblas_operation transa, rocblas_operation transb,
    rocblas_int m, rocblas_int n, rocblas_int k,
    const float *alpha,
    const float *A, rocblas_int lda,
    const float *B, rocblas_int ldb,
    const float *beta,
    float *C, rocblas_int ldc);

该函数的参数含义如下：

- `handle`：rocBLAS库句柄，用于管理rocBLAS库的资源。
- `transa`：矩阵A是否需要转置，可选值为`ROCBLAS_OP_N`和`ROCBLAS_OP_T`。
- `transb`：矩阵B是否需要转置，可选值为`ROCBLAS_OP_N`和`ROCBLAS_OP_T`。
- `m`：矩阵C的行数。
- `n`：矩阵C的列数。
- `k`：矩阵A和B的公共维度。
- `alpha`：矩阵A和B相乘的系数。
- `A`：存储矩阵A的指针。
- `lda`：矩阵A的列数。
- `B`：存储矩阵B的指针。
- `ldb`：矩阵B的列数。
- `beta`：矩阵C的系数。
- `C`：存储矩阵C的指针。
- `ldc`：矩阵C的列数。

该函数实现的矩阵乘法运算为：

C = alpha * A * B + beta * C

其中，矩阵A的维度为m×k，矩阵B的维度为k×n，矩阵C的维度为m×n。

需要注意的是，在调用该函数时，需要保证矩阵A、B、C在设备端（device）上已经分配好了内存，并且矩阵元素的存储顺序是列主序（column-major）存储方式。此外，还需要设置矩阵A、B、C的转置情况，以及矩阵A和B相乘的系数和矩阵C的系数。