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