fix errors caused by gpu in conditions (PaddlePaddle#75551)

Author: YqGe585

python/paddle/distributed/fleet/recompute/recompute.py

@@ -411,7 +411,7 @@ def _recompute_without_reentrant(
 
     if preserve_rng_state:
         cur_device = paddle.get_device()
-        if 'gpu:' in cur_device:
+        if cur_device.startswith('gpu:'):
             fw_cuda_rng_state = paddle.get_cuda_rng_state()
         elif 'cpu' in cur_device:
             fw_cuda_rng_state = paddle.get_rng_state()

python/paddle/incubate/jit/inference_decorator.py

@@ -393,7 +393,7 @@ def create_predictor(self, input_tensor_lists):
         config.enable_new_ir(self.enable_new_ir)
 
         device_num = paddle.device.get_device()
-        if 'gpu' in device_num:
+        if device_num.startswith('gpu'):
             gpu_id = int(device_num.split(':')[1])
             config.enable_use_gpu(
                 self.memory_pool_init_size_mb,

python/paddle/tensor/linalg.py

@@ -4558,7 +4558,7 @@ def lstsq(
                 f"Only support valid driver is 'gels', 'gelss', 'gelsd', 'gelsy' or None for CPU inputs. But got {driver}"
             )
         driver = "gelsy" if driver is None else driver
-    elif "gpu" in device:
+    elif device.startswith('gpu'):
         if driver not in (None, "gels"):
             raise ValueError(
                 f"Only support valid driver is 'gels' or None for CUDA inputs. But got {driver}"

test/legacy_test/test_compat_slogdet.py

@@ -56,7 +56,10 @@ def slogdet_backward(self, x, _, grad_logabsdet):
 
     def test_compat_slogdet(self):
         devices = [paddle.device.get_device()]
-        if "gpu:" in devices and not paddle.device.is_compiled_with_rocm():
+        if (
+            any(device.startswith("gpu:") for device in devices)
+            and not paddle.device.is_compiled_with_rocm()
+        ):
             devices.append("cpu")
         for device in devices:
             with paddle.device.device_guard(device), dygraph_guard():

test/legacy_test/test_div_op.py

@@ -737,7 +737,10 @@ def test_gpu(self):
 
     def test_infer_symbolic_shape(self):
         devices = [paddle.device.get_device()]
-        if "gpu:" in devices and not paddle.device.is_compiled_with_rocm():
+        if (
+            any(device.startswith("gpu:") for device in devices)
+            and not paddle.device.is_compiled_with_rocm()
+        ):
             devices.append("cpu")
 
         for device in devices:

test/legacy_test/test_random_op.py

@@ -116,7 +116,10 @@ def test_random_update_to(self):
 
     def test_pir_random_(self):
         devices = [paddle.device.get_device()]
-        if "gpu:" in devices and not paddle.device.is_compiled_with_rocm():
+        if (
+            any(device.startswith("gpu:") for device in devices)
+            and not paddle.device.is_compiled_with_rocm()
+        ):
             devices.append("cpu")
         for device in devices:
             with paddle.device.device_guard(device), dygraph_guard():

test/sot/test_sot_place.py

@@ -41,7 +41,7 @@ def run_diff_logic_by_check_expected_place(x: paddle.Tensor):
     expected_place_str = paddle.get_device()
     if "cpu" in expected_place_str:
         return x + 1
-    elif "gpu" in expected_place_str:
+    elif expected_place_str.startswith("gpu"):
         return x + 2
     elif "xpu" in expected_place_str:
         return x + 3