From 44231d3911c9bbf8dbcfd9cc15105d84780dec66 Mon Sep 17 00:00:00 2001 From: Nyakku Shigure Date: Wed, 17 Jan 2024 10:35:46 +0800 Subject: [PATCH] [Dy2St] Increase `test_transformer` and `test_mobile_net` ut time (#60829) --- test/dygraph_to_static/CMakeLists.txt | 5 +++-- test/dygraph_to_static/test_mobile_net.py | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt index bddf91c865b14..93983b64f7f05 100644 --- a/test/dygraph_to_static/CMakeLists.txt +++ b/test/dygraph_to_static/CMakeLists.txt @@ -30,6 +30,7 @@ if(NOT WITH_GPU) list(REMOVE_ITEM TEST_OPS test_build_strategy) list(REMOVE_ITEM TEST_OPS test_bert) list(REMOVE_ITEM TEST_OPS test_transformer) + list(REMOVE_ITEM TEST_OPS test_mobile_net) endif() foreach(TEST_OP ${TEST_OPS}) @@ -39,7 +40,6 @@ endforeach() set_tests_properties(test_se_resnet PROPERTIES TIMEOUT 900) set_tests_properties(test_yolov3 PROPERTIES TIMEOUT 900 LABELS "RUN_TYPE=EXCLUSIVE") -set_tests_properties(test_mobile_net PROPERTIES TIMEOUT 120) set_tests_properties(test_seq2seq PROPERTIES TIMEOUT 420) set_tests_properties(test_cycle_gan PROPERTIES TIMEOUT 150) set_tests_properties(test_basic_api_transformation PROPERTIES TIMEOUT 240) @@ -52,13 +52,14 @@ endif() if(APPLE) set_tests_properties(test_bmn PROPERTIES TIMEOUT 300) - set_tests_properties(test_mobile_net PROPERTIES TIMEOUT 300) endif() if(WITH_GPU) set_tests_properties(test_train_step_resnet18_sgd PROPERTIES TIMEOUT 240) set_tests_properties(test_train_step_resnet18_adam PROPERTIES TIMEOUT 240) set_tests_properties(test_bert PROPERTIES TIMEOUT 240) + set_tests_properties(test_transformer PROPERTIES TIMEOUT 240) + set_tests_properties(test_mobile_net PROPERTIES TIMEOUT 240) endif() # Legacy IR only tests for dygraph_to_static diff --git a/test/dygraph_to_static/test_mobile_net.py b/test/dygraph_to_static/test_mobile_net.py index b7be4a68c2752..26b5dc3574848 100644 --- a/test/dygraph_to_static/test_mobile_net.py +++ b/test/dygraph_to_static/test_mobile_net.py @@ -22,7 +22,7 @@ from dygraph_to_static_utils import ( Dy2StTestBase, enable_to_static_guard, - test_legacy_and_pt_and_pir, + test_legacy_and_pir, ) from predictor_utils import PredictorTools @@ -738,7 +738,7 @@ def assert_same_predict(self, model_name): err_msg=f'inference_pred_res:\n {predictor_pre}\n, st_pre: \n{st_pre}.', ) - @test_legacy_and_pt_and_pir + @test_legacy_and_pir def test_mobile_net(self): # MobileNet-V1 self.assert_same_loss("MobileNetV1")