cherry-pick #54567 (#54694)

* cherry-pick #54567

* fix WITH_NV_JETSON TIMEOUT

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)