cherry-pick #54567 (#54694)

* cherry-pick #54567 * fix WITH_NV_JETSON TIMEOUT

Author: LokeZhou

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,20 +950,31 @@ 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)
  set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 1500)
 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)