如何设置cuFFT超时？实现长时间CUDA FFT计算的中断

针对大规模3D FFT偶尔超时的问题，CUDA API本身没有直接的超时设置，但可以通过以下几种方式实现主动中断：

1. 多线程+流/事件监控中断 #

• 启动FFT计算时，同时开启一个独立的监控线程，记录任务开始时间
• 监控线程按固定间隔（如10秒）调用cudaStreamQuery(fft_stream)查询FFT所在流的状态：
  • 返回cudaSuccess则说明任务完成，监控线程退出
  • 若超过设定的超时阈值（如30分钟），调用cudaStreamAbort(fft_stream, cudaErrorAborted)终止流上所有未完成操作（该函数需CUDA 11.0及以上版本）
• 后续FFT相关API会返回cudaErrorAborted错误，捕获该错误并做好资源释放即可

2. 旧版本CUDA的降级方案（CUDA <11.0） #

如果使用低于11.0的CUDA版本，无法使用cudaStreamAbort，可采用：

• 监控线程超时后调用cudaDeviceReset()重置整个GPU设备
• 注意：该操作会清除所有GPU内存中的数据，包括输入的复数数组，需提前做好数据备份或准备重新初始化设备状态

3. 映射内存的配套处理 #

由于你的输出是RAM映射的GPU内存，中断后需要额外处理：

• 调用cudaUnmapResource()释放映射的内存区域，避免内存泄漏
• 重新初始化流或设备后，再重新创建映射资源

极端超时案例日志 #

2023-02-17 21:41:02.174 FFT<C,R> size [960, 1125, 480]
2023-02-18 09:24:46.503 FFT<C,R> complete

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