Fix two tests for custom device (#76116)

* fix some tests

Author: 1184319564

test/legacy_test/test_rnn_decode_api.py

@@ -16,7 +16,7 @@
 import unittest
 
 import numpy as np
-from op_test import is_custom_device
+from op_test import get_device_place, is_custom_device
 
 import paddle
 from paddle import Model, base, nn, set_device
@@ -337,11 +337,10 @@ def check_output_with_place(self, place, mode="test"):
                 )
 
     def check_output(self):
-        devices = (
-            ["CPU", "GPU"]
-            if (base.is_compiled_with_cuda() or is_custom_device())
-            else ["CPU"]
-        )
+        devices = ["CPU"]
+        if base.is_compiled_with_cuda() or is_custom_device():
+            devices.append(get_device_place())
+
         for device in devices:
             place = set_device(device)
             self.check_output_with_place(place)

test/legacy_test/test_set_value_op.py

@@ -1801,15 +1801,15 @@ def test_value_input_is_scalar(self):
 )
 class TestSetValueWithStrideError(unittest.TestCase):
     def test_same_place(self):
-        x = paddle.rand([5, 10], device=paddle.CUDAPlace(0))
-        y = paddle.rand([10, 5], device=paddle.CUDAPlace(0))
+        x = paddle.rand([5, 10], device=get_device_place())
+        y = paddle.rand([10, 5], device=get_device_place())
         y.transpose_([1, 0])
         x.set_value(y)
         assert x.is_contiguous()
 
     def test_different_place1(self):
         # src place != dst place && src is not contiguous
-        x = paddle.rand([5, 10], device=paddle.CUDAPlace(0))
+        x = paddle.rand([5, 10], device=get_device_place())
         y = paddle.rand([10, 5], device=paddle.CPUPlace())
         y.transpose_([1, 0])
         x.set_value(y)
@@ -1818,7 +1818,7 @@ def test_different_place1(self):
     def test_different_place2(self):
         # src place != dst place && dst is not contiguous
         with self.assertRaises(SystemError):
-            x = paddle.ones([5, 4], device=paddle.CUDAPlace(0))
+            x = paddle.ones([5, 4], device=get_device_place())
             x.transpose_([1, 0])
             y = paddle.rand([4, 2], device=paddle.CPUPlace())
             assert not x.is_contiguous()