neurokernel.tools.gpu.set_realloc¶
-
neurokernel.tools.gpu.
set_realloc
(x_gpu, data)[source]¶ Transfer data into a GPUArray instance.
Copies the contents of a numpy array into a GPUArray instance. If the array has a different type or dimensions than the instance, the GPU memory used by the instance is reallocated and the instance updated appropriately.
Parameters: - x_gpu (pycuda.gpuarray.GPUArray) – GPUArray instance to modify.
- data (numpy.ndarray) – Array of data to transfer to the GPU.
Examples
>>> import pycuda.gpuarray as gpuarray >>> import pycuda.autoinit >>> import numpy as np >>> import misc >>> x = np.asarray(np.random.rand(5), np.float32) >>> x_gpu = gpuarray.to_gpu(x) >>> x = np.asarray(np.random.rand(10, 1), np.float64) >>> set_realloc(x_gpu, x) >>> np.allclose(x, x_gpu.get()) True