如何输出CUDA矩阵运算中触发逻辑错误的函数调用所在代码行

核心思路是通过预处理器宏隐式注入调用位置的__FILE__、__LINE__内置宏，无需用户手动修改加法调用代码，就能拿到用户侧的错误位置信息。

具体实现步骤 #

1. 修改Matrix.cuh头文件 #

新增维度校验函数和加法方法声明，添加宏替换逻辑：

// 新增头文件依赖
#include <cstdio>
#include <cstdlib>

// 维度校验错误输出函数
inline void checkDimensionValid(int aRows, int aCols, int bRows, int bCols, const char* file, int line) {
    if (aRows != bRows || aCols != bCols) {
        fprintf(stderr, "Error in file \"%s\" on line: %d (Dimensions inconsistent)\n", file, line);
        exit(EXIT_FAILURE);
    }
}

class Matrix {
    // 原有类成员、方法声明保持不变
public:
    // 新增带位置参数的加法方法，替换原来的operator+
    Matrix& add(const Matrix& B, const char* file, int line);
};

// 核心宏：自动将所有Matrix加法调用替换为带位置参数的add方法
#define operator+(other) add(other, __FILE__, __LINE__)

2. 修改Matrix.cu中的加法实现 #

删除原有的operator+方法，替换为add方法实现，开头加维度校验：

Matrix& Matrix::add(const Matrix& B, const char* file, int line)
{
    // 最先执行维度校验，直接使用用户调用位置的文件、行号信息
    checkDimensionValid(RowCount, ColumnCount, B.RowCount, B.ColumnCount, file, line);

    // 原有加法逻辑保持不变
    Matrix C = sum(B);
    Matrix* C_p;
    cudaMallocManaged(&C_p, sizeof(Matrix));
    memcpy(C_p, &C, sizeof(Matrix));
    B.removePointers();
    C.removePointers();
    return *C_p;
}

实现效果 #

完全不需要修改你现有Main.cu的代码，运行后遇到1010和1515矩阵相加的场景，会直接输出你期望的错误提示：

Error in file "Main.cu" on line: 9 (Dimensions inconsistent)

可选优化建议 #

• 现有operator+返回堆分配的Matrix引用存在内存泄漏风险，建议改为返回值对象，或引入RAII机制管理托管内存
• 维度校验放在Host端执行即可，不需要放到CUDA Kernel中，能提前拦截错误减少不必要的设备调用开销
• sum方法中不需要重复申请三个Matrix类型的托管内存，可直接传递成员变量参数减少内存开销

内容的提问来源于stack exchange，提问作者Thomas Marsden