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-08-18 00:10:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

Create simple_scankernel0.py

Browse Source
This commit is contained in:
Brian Tuomanen
2020-02-18 12:21:48 -08:00
committed by GitHub
parent 217a264a42
commit c91501e0fc
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Chapter03/simple_scankernel0.py Normal file
View File

@@ -0,0 +1,10 @@
import numpy as np
import pycuda.autoinit
from pycuda import gpuarray
from pycuda.scan import InclusiveScanKernel
seq = np.array([1,2,3,4],dtype=np.int32)
seq_gpu = gpuarray.to_gpu(seq)
sum_gpu = InclusiveScanKernel(np.int32, "a+b")
print(sum_gpu(seq_gpu).get())
print(np.cumsum(seq))