在 PyCUDA 中打印消息

Question

在简单的 CUDA 程序中，我们可以通过包含 cuPrintf.h 来按线程打印消息，但是在 PyCUDA 中这样做并没有在任何地方解释。如何在 PyCUDA 中做到这一点？

Answer 1

在 Compute Capability 2.0 及更高版本的 GPU 上，cuPrintf.h 不鼓励使用 CUDA 的内置 printf()。要使用它，只需 #include <stdio.h> 并调用 printf() 就像在主机上一样。

PyCUDA wiki 有一个 specific example of this。