Merge pull request #580 from mrava87/bug-to_cupy_cond

bug: to_cupy_conditional
PyLops · May 22, 2024 · 27636e5 · 27636e5
2 parents 23c5539 + 60997e0
commit 27636e5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pylops/utils/backend.py b/pylops/utils/backend.py
@@ -461,5 +461,6 @@ def to_cupy_conditional(x: npt.ArrayLike, y: npt.ArrayLike) -> NDArray:
     """
     if deps.cupy_enabled:
         if cp.get_array_module(x) == cp and cp.get_array_module(y) == np:
-            y = cp.asarray(y)
+            with cp.cuda.Device(x.device):
+                y = cp.asarray(y)
     return y