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.

Sign up for GitHub

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

Jump to bottom

[Dy2St] Disable test_bert on CPU #60173

Merged
merged 5 commits into from
Dec 21, 2023
Merged

[Dy2St] Disable test_bert on CPU #60173

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
6722eb5
ci test bert
gouzil Dec 20, 2023
bc1b5ad
test
gouzil Dec 20, 2023
f42b2e0
disable cpu
gouzil Dec 20, 2023
f3fff3f
fix
gouzil Dec 20, 2023
92482a0
rename
gouzil Dec 20, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions test/dygraph_to_static/CMakeLists.txt
View file
Open in desktop
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