Allocating memory on the GPU and copying data with PyCUDA (pt. 2)

pycuda_exampple_pt2.py

# create Python variables
a = np.random.randn(100).astype(np.float32)	 
b = np.random.randn(100).astype(np.float32)
result = np.random.randn(100).astype(np.float32)

# allocate memory on GPU
a_gpu = cuda.mem_alloc(a.nbytes)		
b_gpu = cuda.mem_alloc(b.nbytes)
result_gpu = cuda.mem_alloc(b.nbytes)

# copy data to GPU
cuda.memcpy_htod(a_gpu, a)		
cuda.memcpy_htod(b_gpu, b)