Skip to content
Snippets Groups Projects

Replace PyCuda with CuPy

Merged Markus Holzer requested to merge holzer/pystencils:cupy into master
Viewing commit 72aeea14
Show latest version
4 files
+ 6
18
Preferences
Compare changes
Files
4
@@ -116,7 +116,7 @@ class BoundaryHandling:
@@ -116,7 +116,7 @@ class BoundaryHandling:
for obj, cpu_arr in cpu_version.items():
for obj, cpu_arr in cpu_version.items():
if obj not in gpu_version or gpu_version[obj].shape != cpu_arr.shape:
if obj not in gpu_version or gpu_version[obj].shape != cpu_arr.shape:
gpu_version[obj] = array_handler.zeros(cpu_arr.shape, cpu_arr.dtype)
gpu_version[obj] = array_handler.empty(cpu_arr.shape, cpu_arr.dtype)
array_handler.upload(gpu_version[obj], cpu_arr)
array_handler.upload(gpu_version[obj], cpu_arr)
else:
else:
array_handler.upload(gpu_version[obj], cpu_arr)
array_handler.upload(gpu_version[obj], cpu_arr)