Skip to content

Commit 27636e5

Browse files
authored
Merge pull request #580 from mrava87/bug-to_cupy_cond
bug: to_cupy_conditional
2 parents 23c5539 + 60997e0 commit 27636e5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pylops/utils/backend.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -461,5 +461,6 @@ def to_cupy_conditional(x: npt.ArrayLike, y: npt.ArrayLike) -> NDArray:
461461
"""
462462
if deps.cupy_enabled:
463463
if cp.get_array_module(x) == cp and cp.get_array_module(y) == np:
464-
y = cp.asarray(y)
464+
with cp.cuda.Device(x.device):
465+
y = cp.asarray(y)
465466
return y

0 commit comments

Comments
 (0)