PyCUDA 1

PyCUDA

$ tar xfz pycuda-VERSION.tar.gz

Here is what the above code is Doing:
1. Importing the pycuda.driver and pycuda.autoinit modules
2. Loading the kernel code from the file vector_add.cu
3. Compiling the kernel code using the compiler from the CUDA toolkit
4. Allocating some memory on the device
5. Copying some data from the host to the device
6. Executing the kernel
7. Copying the result back from the device
8. Printing the result

Similar Posts