如何设置cuFFT超时?

问题描述 投票:0回答:1

我正在寻找一种方法来中断 cuda FFT 计算(如果它运行时间太长)。怎么才能实现呢?

我在API中寻找一些超时设置,但没有找到这样的选项。在谷歌搜索时,大多数点击都提到 GPU 驱动程序出现不必要的超时,这是一个完全不同的主题。

我正在计算相当大的 3D FFT(大小约为 [1k,1k,0.5k])。通常需要几分钟才能完成。然而,有时由于某些未知原因可能需要几个小时。我在日志中发现的更极端的情况之一是:

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

在这种特殊情况下,输入复数数组位于 GPU 上,但输出实数数组是从 RAM 映射的。否则它可能不适合运行 RTX 3060 Ti 的 8GB 内存。尽管需要绘制地图,但在大多数情况下几分钟就足以完成任务。

cuda cufft
1个回答
0
投票

如评论中所示,目前这是不可能的。

NVIDIA 团队之一建议您的极端示例可能是暴露了 cuFFT 中的错误,应通过 NVIDIA 错误报告系统进行报告。

© www.soinside.com 2019 - 2024. All rights reserved.