cuda - 设备内存刷新cuda

Question

我正在运行一个 C 程序，我在其中调用了两次 cuda 主机函数。我想清理这两个调用之间的设备内存。有什么方法可以刷新 GPU 设备内存？我使用的是计算能力为 2.0 的 Tesla M2050

Answer 1

如果您只想将内存归零，那么 cudaMemset 可能是执行此操作的最简单方法。例如:

const int n = 10000000;
const int sz = sizeof(float) * n;
float *devicemem;
cudaMalloc((void **)&devicemem, sz);

kernel<<<...>>>(devicemem,....);
cudaMemset(devicemem, 0, sz); // zeros all the bytes in devicemem
kernel<<<...>>>(devicemem,....);

注意 cudaMemset 取的值是一个 byte 值，指定范围内的所有字节都设置为该值，就像标准的 C memset 。如果您有一个特定的 word 值，那么您将需要编写自己的 memset 内核来分配这些值。