From e4401c446260234eff10d118c04187c0d00e3685 Mon Sep 17 00:00:00 2001 From: LokeZhou Date: Fri, 16 Jun 2023 17:55:27 +0800 Subject: [PATCH] cherry-pick #54567 (#54694) * cherry-pick #54567 * fix WITH_NV_JETSON TIMEOUT --- test/legacy_test/CMakeLists.txt | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index bd9ad353f79..bd46190cf0b 100644 --- a/test/legacy_test/CMakeLists.txt +++ b/test/legacy_test/CMakeLists.txt @@ -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) @@ -949,7 +950,6 @@ 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) @@ -957,12 +957,24 @@ if(WITH_NV_JETSON) 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) -- GitLab