import pycuda.compiler import SourceModule
import pycuda.autoinit
import pycuda.driver as drv
- mod = SourceModule("“用C语言编写的CUDA程序”")
- func = mod.get_function(“func”)
- 在python语言中调用cuda程序的语法:func(
drv.InOut(a), drv.In(b), N,
block = (nThreads, 1, -1), grid = (nBlocks, 1))