Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Dy2St][PIR] Enable PIR ut test_sentiment and remove test_lac #63192

Merged
merged 14 commits into from
Apr 7, 2024
9 changes: 1 addition & 8 deletions test/dygraph_to_static/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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)
Expand Down
6 changes: 0 additions & 6 deletions test/dygraph_to_static/check_approval.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand Down
Loading