Skip to content

Commit ff2a978

Browse files
committed
test=develop
1 parent dd6fd4c commit ff2a978

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

python/paddle/fluid/__init__.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ def __bootstrap__():
9191
"""
9292
import sys
9393
import os
94+
import platform
9495
from . import core
9596

9697
in_test = 'unittest' in sys.modules
@@ -110,14 +111,17 @@ def __bootstrap__():
110111
print('PLEASE USE OMP_NUM_THREADS WISELY.', file=sys.stderr)
111112

112113
os.environ['OMP_NUM_THREADS'] = str(num_threads)
113-
114+
sysstr = platform.system()
114115
read_env_flags = [
115-
'use_pinned_memory', 'check_nan_inf', 'benchmark', 'eager_delete_scope',
116-
'use_mkldnn', 'use_ngraph', 'initial_cpu_memory_in_mb',
117-
'init_allocated_mem', 'free_idle_memory', 'paddle_num_threads',
118-
"dist_threadpool_size", 'eager_delete_tensor_gb', 'allocator_strategy',
116+
'check_nan_inf', 'benchmark', 'eager_delete_scope', 'use_mkldnn',
117+
'use_ngraph', 'initial_cpu_memory_in_mb', 'init_allocated_mem',
118+
'free_idle_memory', 'paddle_num_threads', "dist_threadpool_size",
119+
'eager_delete_tensor_gb', 'allocator_strategy',
119120
'reader_queue_speed_test_mode', 'print_sub_graph_dir'
120121
]
122+
if sysstr != 'Darwin':
123+
read_env_flags.append('use_pinned_memory')
124+
121125
if os.name != 'nt':
122126
read_env_flags.append('warpctc_dir')
123127
read_env_flags.append('cpu_deterministic')

0 commit comments

Comments
 (0)