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

Question

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

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

Answer 1

即使不查看库头：是的。

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

但是，很有可能有更好的解决方案。我会查看cufftw.h 以了解 nVIDIA 在其中究竟做了什么，以及它究竟是如何“替换”CPU 功能的。