diff --git a/pystencils/datahandling/pycuda.py b/pystencils/datahandling/pycuda.py index 30602a30ce6b87d0e25861b43c5291cda77ed570..75967c1cc25591ba9c4b0b127519205eb2af7208 100644 --- a/pystencils/datahandling/pycuda.py +++ b/pystencils/datahandling/pycuda.py @@ -21,7 +21,7 @@ class PyCudaArrayHandler: def empty(self, shape, dtype=np.float64, layout=None): if layout: cpu_array = pystencils.field.create_numpy_array_with_layout(shape, dtype, layout) - return self.from_numpy(cpu_array) + return self.to_gpu(cpu_array) else: return gpuarray.empty(shape, dtype) @@ -36,4 +36,6 @@ class PyCudaArrayHandler: def randn(self, shape, dtype=np.float64): cpu_array = np.random.randn(*shape).astype(dtype) - return self.from_numpy(cpu_array) + return self.to_gpu(cpu_array) + + from_numpy = to_gpu diff --git a/pystencils/datahandling/pyopencl.py b/pystencils/datahandling/pyopencl.py index 7b6f44088f60c47d0c57c5185f1afd16ef16bac7..b4e53150f498ea2a76ec6ba270e140ea578f4456 100644 --- a/pystencils/datahandling/pyopencl.py +++ b/pystencils/datahandling/pyopencl.py @@ -43,3 +43,5 @@ class PyOpenClArrayHandler: def randn(self, shape, dtype=np.float64): cpu_array = np.random.randn(*shape).astype(dtype) return self.from_numpy(cpu_array) + + from_numpy = to_gpu