From 86858a5acabf666ea163ec1f9ce76f39e4012585 Mon Sep 17 00:00:00 2001 From: LokeZhou Date: Wed, 28 Jun 2023 15:55:07 +0800 Subject: [PATCH] fix test_conv3d_transpose_op A100 test fail (#54913) --- test/legacy_test/CMakeLists.txt | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index 36e71b5ac7f..c81b004f245 100644 --- a/test/legacy_test/CMakeLists.txt +++ b/test/legacy_test/CMakeLists.txt @@ -457,6 +457,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) @@ -950,7 +951,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_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 1500) @@ -960,7 +960,6 @@ 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_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 250) @@ -968,6 +967,18 @@ else() 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) + if(WITH_NV_JETSON) + set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 1200) + else() + set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 120) + endif() +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