如何在 PyCUDA 中使用预处理器符号？

Question

有一个用 C 语言编写的带有一些 CUDA 内核的程序，我需要调整这个程序以使用 PyCUDA 在 python 中运行。现在，在这个 C 代码中定义了一些全局变量，其中一些内核正在通过#ifdef 访问这些变量。我想做的是在 python 中定义这些全局变量，然后复制我的 CUDA C 内核代码并使用 SourceModule 方案运行它们（假设我只是懒惰）。我想像这样：

my_global_var=True
mod=SourceModule(""" __global__ void func() {
...
#ifdef my_global_var
do something
#endif }
""")

有没有办法做到这一点？显然，这种天真的尝试根本行不通。我应该使用某种特殊的指针或类似的东西吗？ 换句话说，当使用 PyCUDA 编译的 CUDA C 内核中有 #ifdef 时会发生什么？

编辑： 好吧，我可能在这里误用了“全局变量”一词。 C 代码看起来或多或少是这样的：

#define X
__global__ void func(...) {
...
#ifdef X
do something
#endif }

我想要实现的是在 PyCUDA 中使用该 C 代码，而不会过多地弄乱它。所以我想复制“global_ void func...”，把它放在 PyCUDA SourceModule 中并在 python 中定义 X（不管它是全局变量还是其他东西）（就像在第一个伪代码）。有可能吗？

Answer 1

与编译任何 C 或 C++ 代码一样，预处理器符号的外部操作（我的意思是在代码本身之外）需要将额外的参数传递给编译器。如果你有这个代码：

__global__ void func(...) {
...
#ifdef X
do something
#endif 

}

那么在编译过程中设置X的方法是将-DX传递给编译器，编译器在当前预处理过程中定义符号X。

您可以在 PyCUDA 中执行完全相同的操作，方法是使用 SourceModule 构造函数中的 options 关键字将选项传递给构建。编译器选项作为列表传递，因此您的 PyCUDA 示例可以这样实现：

my_global_var=True

....

build_options = []
if my_global_var:
    build_options.append('-Dmy_global_var')

mod=SourceModule(""" __global__ void func() {
...
#ifdef my_global_var
do something
#endif }
""", options=build_options)

[免责声明：完全未经测试，使用风险自负 - 我没有工作的 PyCUDA 安装 ATM]

在这里，我们简单地构建一个列表，其中包含与您需要传递给编译器的选项一样多的条目，然后使用这些选项实例化一个 SourceModule 实例。之后，它应该就可以工作了……