From f101defaf83911fc339d25f3700bb21c3c1abb8f Mon Sep 17 00:00:00 2001 From: LokeZhou Date: Tue, 13 Jun 2023 11:04:27 +0000 Subject: [PATCH 1/2] cherry-pick #54567 --- test/legacy_test/CMakeLists.txt | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index bd9ad353f799b..30c4d1e65ab8c 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,21 @@ 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) +else() + py_test_modules(test_conv3d_transpose_op MODULES test_conv3d_transpose_op) + set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 1200) +endif() + set_tests_properties(test_imperative_selected_rows_to_lod_tensor PROPERTIES TIMEOUT 200) set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120) From 6051a1318f24cfe17ffaab80f2d12ba62e0bcf00 Mon Sep 17 00:00:00 2001 From: LokeZhou Date: Thu, 15 Jun 2023 11:39:38 +0000 Subject: [PATCH 2/2] fix WITH_NV_JETSON TIMEOUT --- test/legacy_test/CMakeLists.txt | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index 30c4d1e65ab8c..bd46190cf0be2 100644 --- a/test/legacy_test/CMakeLists.txt +++ b/test/legacy_test/CMakeLists.txt @@ -967,9 +967,12 @@ 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) -else() +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