pycuda 在SourceModule代码段中使用printf

当初想调个代码，没有printf打印，可是难过了半天，最后还是同事翻了墙才找到了答案。

结果超级超级简单，添加#include <stdio.h>

详见样例：

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
mod = SourceModule("""
    #include <stdio.h>
    __global__ void say_hi()
    {
      printf("I am %d.%d\\n", threadIdx.x, threadIdx.y);
    }
    """)
func = mod.get_function("say_hi")

func(block=(4,4,1))

是不是不太相信，来吧，官网demo   UsingPrintf   

啦啦啦啦