Skip to content
Snippets Groups Projects
Commit d49f555a authored by Martin Bauer's avatar Martin Bauer
Browse files

Merge branch 'fix-arrayhandlers' into 'master'

Fix {PyCuda,PyOpenCl}ArrayHandler.{zeros,ones,randn}

See merge request pycodegen/pystencils!125
parents d0064ffb 4dcc3ea5
No related branches found
No related tags found
1 merge request!125Fix {PyCuda,PyOpenCl}ArrayHandler.{zeros,ones,randn}
Pipeline #21056 passed
...@@ -21,7 +21,7 @@ class PyCudaArrayHandler: ...@@ -21,7 +21,7 @@ class PyCudaArrayHandler:
def empty(self, shape, dtype=np.float64, layout=None): def empty(self, shape, dtype=np.float64, layout=None):
if layout: if layout:
cpu_array = pystencils.field.create_numpy_array_with_layout(shape, dtype, 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: else:
return gpuarray.empty(shape, dtype) return gpuarray.empty(shape, dtype)
...@@ -36,4 +36,6 @@ class PyCudaArrayHandler: ...@@ -36,4 +36,6 @@ class PyCudaArrayHandler:
def randn(self, shape, dtype=np.float64): def randn(self, shape, dtype=np.float64):
cpu_array = np.random.randn(*shape).astype(dtype) cpu_array = np.random.randn(*shape).astype(dtype)
return self.from_numpy(cpu_array) return self.to_gpu(cpu_array)
from_numpy = to_gpu
...@@ -43,3 +43,5 @@ class PyOpenClArrayHandler: ...@@ -43,3 +43,5 @@ class PyOpenClArrayHandler:
def randn(self, shape, dtype=np.float64): def randn(self, shape, dtype=np.float64):
cpu_array = np.random.randn(*shape).astype(dtype) cpu_array = np.random.randn(*shape).astype(dtype)
return self.from_numpy(cpu_array) return self.from_numpy(cpu_array)
from_numpy = to_gpu
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment