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
11 lines
286 B
Python
11 lines
286 B
Python
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))
|