override NVIDIA_TF32_OVERRIDE to avoid accuracy problem in 11.2 and 11.8

PaddlePaddle · Dec 26, 2022 · 96d510f · 96d510f
1 parent a4f911d
commit 96d510f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_einsum_0d_tensor.py b/python/paddle/fluid/tests/unittests/test_einsum_0d_tensor.py
@@ -12,13 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import unittest
 
 import numpy as np
 from numpy.testing import assert_allclose
 
 import paddle
 
+os.environ['NVIDIA_TF32_OVERRIDE'] = "0"
+
 
 class Test0DCase0(unittest.TestCase):
     def setUp(self):