bug: fixes PyLops#579

mrava87 · May 22, 2024 · 60997e0 · 60997e0
1 parent 42c6b2f
commit 60997e0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pylops/utils/backend.py b/pylops/utils/backend.py
@@ -437,5 +437,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