Lerking/Hands-On-GPU-Programming-with-CUDA-C-and-Python-3.x-Second-Edition
1
0
Fork 0
mirror of https://github.com/PacktPublishing/Hands-On-GPU-Programming-with-CUDA-C-and-Python-3.x-Second-Edition.git synced 2025-07-21 21:01:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

Create simple_scalar_multiply_kernel.py

Browse Source
This commit is contained in:
Brian Tuomanen
2020-02-18 12:53:35 -08:00
committed by GitHub
parent 557759227a
commit 6d52f71c26
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
Chapter04/simple_scalar_multiply_kernel.py Normal file
View File

@@ -0,0 +1,23 @@
import pycuda.autoinit
import pycuda.driver as drv
import numpy as np
from pycuda import gpuarray
from pycuda.compiler import SourceModule
ker = SourceModule("""
__global__ void scalar_multiply_kernel(float *outvec, float scalar, float *vec)
{
int i = threadIdx.x;
outvec[i] = scalar*vec[i];
}
""")
scalar_multiply_gpu = ker.get_function("scalar_multiply_kernel")
testvec = np.random.randn(512).astype(np.float32)
testvec_gpu = gpuarray.to_gpu(testvec)
outvec_gpu = gpuarray.empty_like(testvec_gpu)
scalar_multiply_gpu( outvec_gpu, np.float32(2), testvec_gpu, block=(512,1,1), grid=(1,1,1))
print("Does our kernel work correctly? : {}".format(np.allclose(outvec_gpu.get() , 2*testvec) ))