c++ - 有没有办法在一个 C++ 程序中同时使用 FFTW 和 cuFFTW？

Question

我正在编写一段使用 FFTW3 库的代码，并希望使用 CUDA 将它移植到 GPU。最简单的方法是使用 cuFFTW 兼容库，但正如文档所述，它意味着用 GPU 等效项完全替换 FFTW 的 CPU 版本。添加 cufftw.h header 后，它替换了所有 CPU 函数，代码在 GPU 上运行。

但有没有办法在我的代码中同时拥有 FFTW 的 CPU 和 GPU 版本，这样我就可以在运行时在它们之间切换，例如通过用户界面中的复选框？还是唯一的方法是使用 native cuFFT API？

Answer 1

即使不查看库标题:是的。

一个简单的解决方案是让不同的编译单元每个只包含一个库，并且只公开您自己的自定义函数/类/等。到应用程序的其余部分。这样两个库的头文件就不会互相干扰。

但是，很可能有更好的解决方案。我会查看 cufftw.h，看看 nVIDIA 到底在做什么，以及它是如何“取代”CPU 功能的。