Update massage_args method to fix issues with pyopencl

pypr · Feb 16, 2019 · e2d75fc · e2d75fc
1 parent 965d0d7
commit e2d75fc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/compyle/jit.py b/compyle/jit.py
@@ -280,7 +280,7 @@ def _generate_kernel(self, *args):
     def _massage_arg(self, x):
         if isinstance(x, array.Array):
             return x.dev
-        elif isinstance(x, np.ndarray):
+        elif self.backend != 'cuda' or isinstance(x, np.ndarray):
             return x
         else:
             return np.asarray(x)
@@ -353,7 +353,7 @@ def _generate_kernel(self, *args):
     def _massage_arg(self, x):
         if isinstance(x, array.Array):
             return x.dev
-        elif isinstance(x, np.ndarray):
+        elif self.backend != 'cuda' or isinstance(x, np.ndarray):
             return x
         else:
             return np.asarray(x)
@@ -459,7 +459,7 @@ def _generate_kernel(self, **kwargs):
     def _massage_arg(self, x):
         if isinstance(x, array.Array):
             return x.dev
-        elif isinstance(x, np.ndarray):
+        elif self.backend != 'cuda' or isinstance(x, np.ndarray):
             return x
         else:
             return np.asarray(x)