fix: handle multiple GPUs in paddle backend to_device

ivy-llc · Oct 25, 2023 · 40b27d8 · 40b27d8
1 parent 7e9e762
commit 40b27d8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ivy/functional/backends/paddle/device.py b/ivy/functional/backends/paddle/device.py
@@ -34,7 +34,11 @@ def to_device(
     device = as_native_dev(device)
     if device.is_cpu_place() and not x.place.is_cpu_place():
         return x.cpu()
-    elif device.is_gpu_place() and not x.place.is_gpu_place():
+    elif (device.is_gpu_place() and not x.place.is_gpu_place()) or (
+        x.place.is_gpu_place()
+        and device.is_gpu_place()
+        and x.gpu_device_id() != device.gpu_device_id()
+    ):
         return x.cuda(device.gpu_device_id())
     else:
         return x