Skip to content

Commit 945ea69

Browse files
authored
Revert "Disable NVIDIA_TF32_OVERRIDE by default for better precision." (PaddlePaddle#75907)
* Revert "Disable CUBLAS TF32 for default for better precision. (PaddlePaddle#75476)" This reverts commit fcf3c3f. * Update __init__.py test=document_fix
1 parent 5458524 commit 945ea69

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

python/paddle/base/__init__.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,6 @@ def __bootstrap__():
164164

165165
os.environ['OMP_NUM_THREADS'] = str(num_threads)
166166

167-
if os.getenv('NVIDIA_TF32_OVERRIDE', None) is None:
168-
os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
169-
170167
if os.getenv('MKL_NUM_THREADS', None) is None:
171168
os.environ['MKL_NUM_THREADS'] = str(int(0.8 * os.cpu_count()))
172169

0 commit comments

Comments
 (0)