diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt index 36e71b5ac7ffaaf53af0472c5347487eda09f9d4..c81b004f245e3e8213f6f896c497c70774f82021 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)