使用CUDA对矩阵赋值时出现Segmentation Fault问题求助

1. 段错误根因：主机栈溢出 #

你在main函数内声明的栈上二维数组float A[4096][65536]总大小为1GB，而操作系统默认进程栈空间仅为8~32MB，完全无法承载这么大的内存分配，程序在进入main函数阶段就会触发段错误，和你怀疑的CUDA设备到主机拷贝逻辑没有直接关联。

2. 其余潜在问题 #

• 核函数存在越界写入风险：当线程全局索引i >= n*m时，x、y变量未初始化就会执行A[x][y] = value，会破坏显存数据，严重时也会触发程序崩溃。
• 网格尺寸计算错误：ceil((n*m)/block_size)的写法中，n*m和block_size都是整数，执行的是截断整数除法，ceil函数无法发挥向上取整的作用，会导致启动的线程总数不足，矩阵部分元素未被赋值。
• 缺少CUDA调用错误检查：所有CUDA API调用、核函数启动都没有加错误校验，无法快速定位运行时错误。

3. 修复后可运行代码 #

#include <cuda.h>
#include <iostream>

using namespace std;

__global__ void initKernel(float A[][65536], int n, int m, float value){
    int i = blockDim.x*blockIdx.x + threadIdx.x;
    // 只有合法索引才执行写入操作
    if(i < n*m){
        int x = i/m;
        int y = i%m;
        A[x][y] = value;
    }
}

void matrixInit(float A[][65536], int n, int m, float value){
    int size = n * m * sizeof(float);
    int block_size = 32;
    // 正确的向上取整计算网格大小
    int number_of_blocks = (n * m + block_size - 1) / block_size;

    float (*d_A)[65536];
    cudaMalloc((void**)&d_A, size);
    cudaMemcpy(d_A, A, size, cudaMemcpyHostToDevice);
    initKernel<<<number_of_blocks, block_size>>>(d_A, n, m, value);
    // 核函数启动错误检查
    cudaError_t err = cudaGetLastError();
    if (err != cudaSuccess) {
        cerr << "核函数启动失败: " << cudaGetErrorString(err) << endl;
        return;
    }
    cudaDeviceSynchronize();
    cudaMemcpy(A, d_A, size, cudaMemcpyDeviceToHost);
    cudaFree(d_A);
}

int main(){
    int n = 4096;
    int m = 65536;
    // 大数组从栈上移到堆上分配
    float (*A)[65536] = new float[n][65536];
    matrixInit(A, n, m, 1.0);
    // 测试验证第一个和最后一个元素
    cout << "A[0][0] = " << A[0][0] << endl;
    cout << "A[4095][65535] = " << A[4095][65535] << endl;
    delete[] A;
    return 0;
}

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