cuda - 在 PyCUDA 中打印消息

Question

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

Answer 1

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

PyCUDA wiki 有一个具体的例子。