Skip to content
Snippets Groups Projects

Implement Pinned GPU memory

Merged Markus Holzer requested to merge holzer/pystencils:PinnedMem into master
Viewing commit 7e32f00c
Prev
Show latest version
1 file
+ 15
0
Preferences
Compare changes
@@ -387,3 +387,18 @@ def test_array_handler():
random_array = array_handler.randn(size)
cpu_array = np.empty((20, 40), dtype=np.float64)
gpu_array = array_handler.to_gpu(cpu_array)
assert cpu_array.base is None
assert gpu_array.base is None
assert gpu_array.strides == cpu_array.strides
cpu_array2 = np.empty((20, 40), dtype=np.float64)
cpu_array2 = cpu_array2.swapaxes(0, 1)
gpu_array2 = array_handler.to_gpu(cpu_array2)
assert cpu_array2.base is not None
assert gpu_array2.base is not None
assert gpu_array2.strides == cpu_array2.strides