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

cherry-pick #54567 #54694

Merged
merged 2 commits into from
Jun 16, 2023
Merged

cherry-pick #54567 #54694

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
f101def
cherry-pick #54567
LokeZhou Jun 13, 2023
6051a13
fix WITH_NV_JETSON TIMEOUT
LokeZhou Jun 15, 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
16 changes: 14 additions & 2 deletions test/legacy_test/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -456,6 +456,7 @@ list(REMOVE_ITEM TEST_OPS test_basic_lstm_unit_op)
list(REMOVE_ITEM TEST_OPS test_fuse_all_reduce_pass)
list(REMOVE_ITEM TEST_OPS test_fuse_bn_act_pass)
list(REMOVE_ITEM TEST_OPS test_fuse_bn_add_act_pass)
list(REMOVE_ITEM TEST_OPS test_conv3d_transpose_op)

# disable this unittest temporarily
list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
Expand Down Expand Up @@ -949,20 +950,31 @@ endif()
if(WITH_NV_JETSON)
set_tests_properties(test_concat_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_norm_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 1500)
set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 1500)
else()
set_tests_properties(test_concat_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_norm_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 250)
set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150)
endif()

if((WITH_GPU OR WITH_ROCM) AND (LINUX))
py_test_modules(test_conv3d_transpose_op MODULES test_conv3d_transpose_op
ENVS NVIDIA_TF32_OVERRIDE=0)
set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 120)
elseif(WITH_NV_JETSON)
py_test_modules(test_conv3d_transpose_op MODULES test_conv3d_transpose_op)
set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 1200)
else()
py_test_modules(test_conv3d_transpose_op MODULES test_conv3d_transpose_op)
set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 120)
endif()

set_tests_properties(test_imperative_selected_rows_to_lod_tensor
PROPERTIES TIMEOUT 200)
set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120)
Expand Down