revert test_tensor_memcpy_on_cpu

PaddlePaddle · Dec 5, 2023 · 33498cd · 33498cd
1 parent e319064
commit 33498cd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/dygraph_to_static/test_tensor_memcpy_on_cpu.py b/test/dygraph_to_static/test_tensor_memcpy_on_cpu.py
@@ -15,7 +15,11 @@
 import unittest
 
 import numpy as np
-from dygraph_to_static_utils import Dy2StTestBase, test_legacy_and_pt_and_pir
+from dygraph_to_static_utils import (
+    Dy2StTestBase,
+    test_legacy_and_pt,
+    test_legacy_and_pt_and_pir,
+)
 
 import paddle
 
@@ -66,7 +70,7 @@ def _run(self, to_static):
         x2 = paddle.jit.to_static(tensor_copy_to_cuda)(x1)
         return x1.place, x2.place, x2.numpy()
 
-    @test_legacy_and_pt_and_pir
+    @test_legacy_and_pt
     def test_tensor_cuda_on_default_cpu(self):
         if not paddle.is_compiled_with_cuda():
             return