python cuda_Python通过pycuda使用CUDA扩展

python对CUDA扩展有不错的支持，CUDA通过大量线程的并行化可以大幅提高代码计算速度，一般python常用numba、pycuda套件来支持CUDA扩展。numba通过JIT编译器只需将numba装饰器应用到python函数中即可实现CUDA加速，而pycuda需要基于C/C++编写kernel，其移植性、直观性更佳，这里主要介绍pycuda的使用。

1.向量加法

示例使用了1个block，block中含有400个线程，每个线程计算向量加法最终结果的一个值。

import numpy

import pycuda.autoinit

import pycuda.driver as cuda

from pycuda.compiler import SourceModule

mod = SourceModule("""

__global__ void vect_add(float *dest, float *a, float *b)

{

const int i = threadIdx.x;

dest[i] = a[i] + b[i];

}

""")

vect_add = mod.get_function("vect_add")

a = numpy.random.randn(400).astype(numpy.float32)

b = numpy.random.randn(400).astype(numpy.float32)

dest = numpy.zeros_like(a)

vect_add(cuda.Out(dest),