Skip to content

Commit fb4637d

Browse files
authored
[Dy2St] Disable test_bert on CPU (#60173)
1 parent 2e4fa42 commit fb4637d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/dygraph_to_static/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ if(NOT WITH_GPU)
2828
# disable some model test on CPU to avoid timeout
2929
list(REMOVE_ITEM TEST_OPS test_resnet)
3030
list(REMOVE_ITEM TEST_OPS test_build_strategy)
31+
list(REMOVE_ITEM TEST_OPS test_bert)
3132
endif()
3233

3334
foreach(TEST_OP ${TEST_OPS})
@@ -40,12 +41,10 @@ set_tests_properties(test_yolov3 PROPERTIES TIMEOUT 900 LABELS
4041
set_tests_properties(test_mobile_net PROPERTIES TIMEOUT 120)
4142
set_tests_properties(test_seq2seq PROPERTIES TIMEOUT 420)
4243
set_tests_properties(test_cycle_gan PROPERTIES TIMEOUT 150)
43-
set_tests_properties(test_bert PROPERTIES TIMEOUT 180)
4444
set_tests_properties(test_basic_api_transformation PROPERTIES TIMEOUT 240)
4545
set_tests_properties(test_reinforcement_learning PROPERTIES TIMEOUT 120)
4646
set_tests_properties(test_transformer PROPERTIES TIMEOUT 200)
4747
set_tests_properties(test_bmn PROPERTIES TIMEOUT 300)
48-
set_tests_properties(test_bert PROPERTIES TIMEOUT 240)
4948

5049
if(NOT WIN32)
5150
set_tests_properties(test_tsm PROPERTIES TIMEOUT 900)
@@ -59,6 +58,7 @@ endif()
5958
if(WITH_GPU)
6059
set_tests_properties(test_train_step_resnet18_sgd PROPERTIES TIMEOUT 240)
6160
set_tests_properties(test_train_step_resnet18_adam PROPERTIES TIMEOUT 240)
61+
set_tests_properties(test_bert PROPERTIES TIMEOUT 240)
6262
endif()
6363

6464
# Legacy IR only tests for dygraph_to_static

0 commit comments

Comments
 (0)