未验证 提交 e4401c44 编写于 作者: L LokeZhou 提交者: GitHub

cherry-pick #54567 (#54694)

* cherry-pick #54567

* fix WITH_NV_JETSON TIMEOUT
上级 8077d79a
...@@ -456,6 +456,7 @@ list(REMOVE_ITEM TEST_OPS test_basic_lstm_unit_op) ...@@ -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_all_reduce_pass)
list(REMOVE_ITEM TEST_OPS test_fuse_bn_act_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_fuse_bn_add_act_pass)
list(REMOVE_ITEM TEST_OPS test_conv3d_transpose_op)
# disable this unittest temporarily # disable this unittest temporarily
list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception) list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
...@@ -949,7 +950,6 @@ endif() ...@@ -949,7 +950,6 @@ endif()
if(WITH_NV_JETSON) if(WITH_NV_JETSON)
set_tests_properties(test_concat_op PROPERTIES TIMEOUT 1200) 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_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_conv3d_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_norm_op PROPERTIES TIMEOUT 1200) set_tests_properties(test_norm_op PROPERTIES TIMEOUT 1200)
set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 1500) set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 1500)
...@@ -957,12 +957,24 @@ if(WITH_NV_JETSON) ...@@ -957,12 +957,24 @@ if(WITH_NV_JETSON)
else() else()
set_tests_properties(test_concat_op PROPERTIES TIMEOUT 120) 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_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_conv3d_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_norm_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_layer_norm_op PROPERTIES TIMEOUT 250)
set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150) set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150)
endif() 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 set_tests_properties(test_imperative_selected_rows_to_lod_tensor
PROPERTIES TIMEOUT 200) PROPERTIES TIMEOUT 200)
set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120) set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册