【Fix PIR Unittest No.135】fix test_io_save_load and test_jit_save_load

test/dygraph_to_static/test_pylayer.py

@@ -28,7 +28,7 @@
 import unittest
 
 import numpy as np
-from test_jit_save_load_rename import train
+from test_jit_save_load import train
 
 import paddle
 from paddle.autograd.py_layer import PyLayer

test/legacy_test/CMakeLists.txt

@@ -492,7 +492,7 @@ foreach(TEST_OP ${TEST_OPS_WITH_GC})
 endforeach()
 
 # Switch some dy2st UT to eager mode
-set(TEST_EAGER_OPS test_jit_save_load_rename test_translated_layer)
+set(TEST_EAGER_OPS test_jit_save_load test_translated_layer)
 foreach(TEST_OP ${TEST_EAGER_OPS})
   list(REMOVE_ITEM TEST_OPS ${TEST_OP})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS FLAGS_enable_eager_mode=1)
@@ -855,7 +855,7 @@ set_tests_properties(test_vision_models PROPERTIES TIMEOUT 120)
 set_tests_properties(test_dataset_uci_housing PROPERTIES TIMEOUT 120)
 set_tests_properties(test_dataset_imdb PROPERTIES TIMEOUT 300)
 set_tests_properties(test_callback_wandb PROPERTIES TIMEOUT 60)
-set_tests_properties(test_jit_save_load_rename PROPERTIES TIMEOUT 100)
+set_tests_properties(test_jit_save_load PROPERTIES TIMEOUT 100)
 if(WITH_COVERAGE)
   set_tests_properties(test_hapi_hub PROPERTIES TIMEOUT 300)
 endif()

test/deprecated/legacy_test/test_io_save_load.py → test/legacy_test/test_io_save_load.py

@@ -19,7 +19,7 @@
 import paddle
 from paddle import base, static
 from paddle.base import core
-from paddle.pir_utils import test_with_pir_api
+from paddle.pir_utils import test_with_dygraph_pir, test_with_pir_api
 
 
 class TestSaveLoadAPIError(unittest.TestCase):
@@ -97,7 +97,7 @@ def setUp(self):
     def tearDown(self):
         self.temp_dir.cleanup()
 
-    @test_with_pir_api
+    @test_with_dygraph_pir
     def test_when_train_with_no_grad(self):
         paddle.disable_static()
         net = paddle.nn.Linear(1024, 1)