diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt index e2ce58b7cf58c..e9ae745681017 100644 --- a/test/dygraph_to_static/CMakeLists.txt +++ b/test/dygraph_to_static/CMakeLists.txt @@ -28,6 +28,7 @@ if(NOT WITH_GPU) # disable some model test on CPU to avoid timeout list(REMOVE_ITEM TEST_OPS test_resnet) list(REMOVE_ITEM TEST_OPS test_build_strategy) + list(REMOVE_ITEM TEST_OPS test_bert) endif() foreach(TEST_OP ${TEST_OPS}) @@ -40,12 +41,10 @@ set_tests_properties(test_yolov3 PROPERTIES TIMEOUT 900 LABELS 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_bert PROPERTIES TIMEOUT 180) set_tests_properties(test_basic_api_transformation PROPERTIES TIMEOUT 240) set_tests_properties(test_reinforcement_learning PROPERTIES TIMEOUT 120) set_tests_properties(test_transformer PROPERTIES TIMEOUT 200) set_tests_properties(test_bmn PROPERTIES TIMEOUT 300) -set_tests_properties(test_bert PROPERTIES TIMEOUT 240) if(NOT WIN32) set_tests_properties(test_tsm PROPERTIES TIMEOUT 900) @@ -59,6 +58,7 @@ 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) endif() # Legacy IR only tests for dygraph_to_static