第一个PyCUDA程序

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule

mod = SourceModule("""
    #include <stdio.h>

     __global__ void myfirst_kernel()
       {
        printf("Hello,PyCUDA!!!");
      }
""")
 
function = mod.get_function("myfirst_kernel")
function(block=(1,1,1))

开发PyCUDA代码的第一步是导入代码所需的所有库，import 指令用于在文件中导入库、模块、类或函数，类似于在C或C++中的导入指令。可以通过三种不同的方式来完成，如下步骤所示。三个使用导入模块的情况如下。
1 ) lmport pycuda.driver as drv
这表示导入pymodule的驱动子模块，并为其赋予一个简短的符号drv，因此只要使用pycuda.driver模块中的函数，就可以将它们用作 drv.functionname。此模块包含内存管理功能、设备属性、数据方向功能等。
2 ) lmport pycuda.autoinit
此命令指示从pycuda导入 autoint模块，没有任何简短符号。autoint模块用于设备初始化、上下文创建和内存清理。此模块不是强制性的，所有上述功能也可以手动完成。
3 ) From pycuda.compiler import SourceModule
此命令指示只导入pycuda.compiler模块中的SourceModule类。当你只想使用一个大型模块的类时这一点很重要。SourceModule类用于在PyCUDA中定义类C的内核函数。
C或C++内核代码作为构造函数被传送到SourceModule类并创建mod对象。这里的内核代码非常简单，因为它只是在控制台上打印一个Hello,PyCUDA!字符串。由于printf函数在内核代码中使用，所以导入 stdio.h头文件是非常重要。myfirst_kernel 函数在内核代码中定义，使用__global__ 指令，指示将在GPU上执行该函数。函数没有任何参数，它只是在控制台上打印一个字符串。这个内核函数将由nvcc编译器进行编译。
使用mod对象的 get_function方法创建指向函数的指针，然后可以在Python代码中使用此函数。内核函数的名称以引号作为参数提供，指针变量可以被赋予任何名称。这个指针变量用于在代码的最后一行调用内核。可以在此处指定内核函数的参数，由于myfirst_kernel 函数没有任何参数，因此未指定任何参数。通过使用可选的线程块和网格参数，还可以将每个块的线程数和内核启动的每个网格的块数作为参数提供。block参数的值为( 1, 1, 1)，它是1×3的 Python元组，表示块大小为1×1×1。因此将启动一个线程，该线程将在控制台上打印字符串。
使用ANSI C关键字和 CUDA扩展关键字编写的设备代码称为内核，由一个名为内核调用的方式从Python代码启动。简单地说，内核调用的含义是我们从主机代码启动设备代码。内核调用通常会生成大量块和线程来在GPU并行地处理数据。内核代码与普通的C函数非常相似，只是这段代码是由多个线程并行执行的。在Python中有一个非常简单的语法，如下所示:

kernel (parameters for kernel, block=(tx,ty,tz), grid=(bx, by, bz) )

它以我们想要启动的内核函数的指针开始，你必须确保这个内核指针是使用get_function方法创建的。然后可以包含用逗号分隔的内核函数的参数，block参数表示要启动的线程数,Grid参数表示网格中的线程块数。block 和 Grid参数使用1×3的Python元组指定，该元组指示三维中的线程块和线程，内核启动的线程总数是这两个数字的乘积。