cublasStrmm（CUDA三角矩阵乘法）执行结果不符合预期的技术咨询

你好呀！我看到你在使用cuBLAS的cublasStrmm做三角矩阵乘法时，碰到了输出和预期不符的问题，也有可能是对某些参数的理解还没摸透。先把你贴的代码片段整理好（我补了打印换行的部分，方便矩阵可读性）：

#include <cstdio> 
#include <cublas_v2.h> 
#include <cuda_runtime.h>  

void PrintMatrix(const char *name, float *mat, int M, int N) {  
  printf("Matrix: %s\n", name);  
  for (int i = 0; i < M; ++i) {  
    for (int j = 0; j < N; ++j) {  
      printf("%.2f, ", mat[i * N + j]);
    }
    printf("\n"); // 补充换行，让矩阵打印更清晰
  }
}

从目前的代码片段来看，有个很容易踩坑的点要先提醒你：cuBLAS库默认是用列优先存储矩阵的，但你写的PrintMatrix是按行优先的逻辑来打印的！如果你的输入矩阵是按行优先准备的，直接传给cuBLAS函数计算，结果肯定会和你预期的不一样，这是很多新手用cuBLAS都会犯的错。

另外，给你列几个常见的排查方向，你可以逐一核对：

• cublasStrmm的核心参数：比如你设置的是左乘还是右乘三角矩阵（CUBLAS_SIDE_LEFT/CUBLAS_SIDE_RIGHT）、三角矩阵是上三角还是下三角（CUBLAS_FILL_MODE_UPPER/CUBLAS_FILL_MODE_LOWER）、是否是单位三角矩阵（CUBLAS_DIAG_NON_UNIT/CUBLAS_DIAG_UNIT），这些参数哪怕写错一个，结果都会天差地别。
• 内存拷贝的正确性：要确认主机到设备、设备到主机的cudaMemcpy有没有用对方向，比如cudaMemcpyHostToDevice和cudaMemcpyDeviceToHost搞反的话，要么传错数据，要么读回的是垃圾值。
• API错误检查：每次调用cuBLAS或者CUDA的API后，一定要检查返回的错误码！比如cublasStatus_t status = cublasStrmm(...)，然后判断status是不是CUBLAS_STATUS_SUCCESS，很多隐性问题都能通过错误码揪出来。
• 矩阵维度匹配：要确保输入的三角矩阵和待乘矩阵的维度符合要求，比如左乘三角矩阵时，三角矩阵的行数必须和待乘矩阵的行数一致，维度不匹配的话计算结果肯定不对。

如果能把完整的cublasStrmm调用代码、你预期的输出结果，还有实际得到的输出都补充上来，我可以帮你更精准地定位问题哦！

内容来源于stack exchange