linux - 主机进程终止时的 CUDA 上下文销毁

Question

如果我的宿主程序 [exit]/[segfault]/[is killed] 关于 CUDA 上下文破坏和相应分配资源的相应行为是什么？

我所说的“行为”是指如果我从未在我的程序中显式调用 cudaDeviceReset，则自动 GPU 驱动程序端机制。

我在 Linux 下工作，我看到我的 GPU 内存使用量在几天(甚至一天)内逐渐增加，同时我正在启动 CUDA 代码，有时通过 (CTRL-C) 或简单地段错误终止(在主机端)...

Answer 1

也许，如果你正确地管理异常，你可以使用atexit()？使用 atexit() 的示例如下:

void ExitFunction() {
    cudaDeviceReset();
}

int main(...) {
    atexit(ExitFunction); 
    ...
    return 0;
}