Pass NVIDIA_TF32_OVERRIDE to internal (#43646)

python/paddle/fluid/tests/unittests/test_collective_api_base.py

@@ -229,6 +229,10 @@ class TestDistBase(unittest.TestCase):
             required_envs["GLOG_logtostderr"] = "1"
             required_envs["GLOO_LOG_LEVEL"] = "TRACE"
 
+        if os.getenv('NVIDIA_TF32_OVERRIDE', '') is not None:
+            required_envs['NVIDIA_TF32_OVERRIDE'] = os.getenv(
+                'NVIDIA_TF32_OVERRIDE', '')
+
         if eager_mode:
             required_envs["FLAGS_enable_eager_mode"] = "%d" % 1
         else:

python/paddle/fluid/tests/unittests/test_dist_base.py

@@ -1468,6 +1468,10 @@ class TestDistBase(unittest.TestCase):
                 "grpc_server=10,request_handler_impl=10,section_worker=10"
             required_envs["GLOG_logtostderr"] = "1"
 
+        if os.getenv('NVIDIA_TF32_OVERRIDE', '') is not None:
+            required_envs['NVIDIA_TF32_OVERRIDE'] = os.getenv(
+                'NVIDIA_TF32_OVERRIDE', '')
+
         required_envs.update(need_envs)
         return required_envs