Skip to content

Commit fcf3c3f

Browse files
authored
Disable CUBLAS TF32 for default for better precision. (PaddlePaddle#75476)
1 parent cf92c0c commit fcf3c3f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/paddle/base/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,9 @@ 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+
167170
flag_prefix = "FLAGS_"
168171
read_env_flags = [
169172
key[len(flag_prefix) :]

0 commit comments

Comments
 (0)