[Dy2St][PIR] Enable PIR ut test_sentiment and remove test_lac (Pa…

…ddlePaddle#63192)
co63oc · Apr 10, 2024 · efc7ec4 · efc7ec4
1 parent b8e9ea5
commit efc7ec4
Show file tree

Hide file tree

Showing 5 changed files with 74 additions and 742 deletions.
diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt
@@ -7,14 +7,6 @@ set(SOT_ENVS SOT_LOG_LEVEL=0 COST_MODEL=False MIN_GRAPH_SIZE=0
              STRICT_MODE=False)
 set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0)
 
-list(REMOVE_ITEM TEST_OPS test_lac)
-# NOTE(Aurelius84): In case of Windows CI, if open ON_INFER, RWLOCK of Scope
-# will be removed and will cause some random failed in multi-thread.
-if(WITH_PYTHON)
-  py_test_modules(test_lac MODULES test_lac)
-  set_tests_properties(test_lac PROPERTIES TIMEOUT 120)
-endif()
-
 if(WIN32 AND NOT WITH_GPU)
   # disable on Windows CPU CI for timeout
   list(REMOVE_ITEM TEST_OPS test_resnet_amp)
@@ -47,6 +39,7 @@ set_tests_properties(test_reinforcement_learning PROPERTIES TIMEOUT 120)
 set_tests_properties(test_bmn PROPERTIES TIMEOUT 300)
 set_tests_properties(test_loop PROPERTIES TIMEOUT 180)
 set_tests_properties(test_mnist_amp PROPERTIES TIMEOUT 240)
+set_tests_properties(test_sentiment PROPERTIES TIMEOUT 150)
 
 if(TEST test_resnet_amp)
   set_tests_properties(test_resnet_amp PROPERTIES TIMEOUT 360)

diff --git a/test/dygraph_to_static/check_approval.py b/test/dygraph_to_static/check_approval.py
@@ -130,7 +130,6 @@ def __init__(self, start: Location, end: Location):
         "test_error.py",
         "test_op_attr.py",
         "test_se_resnet.py",
-        "test_lac.py",
         "test_convert_call.py",
         "test_local_cast.py",
         "test_origin_info.py",
@@ -156,18 +155,13 @@ def __init__(self, start: Location, end: Location):
         "test_logical.py",
         "test_inplace_assign.py",
         # TODO: Remove these files from the allow list after it's support PIR mode
-        "test_list.py",
         "test_bmn.py",
         "test_tensor_hook.py",
-        "test_container.py",
         "test_to_tensor.py",
         "test_warning.py",
         "test_typing.py",
         "test_gradname_parse.py",
-        "test_cache_program.py",
         "test_for_enumerate.py",
-        "test_lac.py",
-        "test_sentiment.py",
         "test_save_load.py",
         "test_cinn.py",
         "test_declarative.py",