Skip to content

Commit

Permalink
[Dy2St] Disable test_bert on CPU (PaddlePaddle#60173) (PaddlePaddle…
Browse files Browse the repository at this point in the history
…#60324)

Co-authored-by: gouzil <66515297+gouzil@users.noreply.github.com>
  • Loading branch information
SigureMo and gouzil authored Dec 26, 2023
1 parent 1b696a1 commit a4cd847
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/dygraph_to_static/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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})
Expand All @@ -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)
Expand All @@ -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
Expand Down

0 comments on commit a4cd847

Please sign in to comment.