Skip to content

Commit 3e9be64

Browse files
committed
Remove OMP_THREADS env variable
1 parent 9d28e8a commit 3e9be64

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

runner.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -166,12 +166,7 @@ def is_ht_enabled():
166166
hostname = socket.gethostname()
167167

168168
cpu_count = multiprocessing.cpu_count()
169-
if is_ht_enabled():
170-
omp_num_threads = str(cpu_count // 2)
171-
omp_places = f'0:{cpu_count}:1'
172-
else:
173-
omp_num_threads = str(cpu_count)
174-
omp_places = ''
169+
omp_num_threads = str(cpu_count // 2) if is_ht_enabled() else str(cpu_count)
175170

176171
# get parameters that are common for all cases
177172
common_params = config['common']
@@ -267,7 +262,6 @@ class GenerationArgs:
267262
env = os.environ.copy()
268263
if lib == 'xgboost':
269264
env['OMP_NUM_THREADS'] = omp_num_threads
270-
env['OMP_PLACES'] = omp_places
271265

272266
for i, case in enumerate(cases):
273267
command = f'python {lib}/{algorithm}.py --batch {batch} ' \

0 commit comments

Comments
 (0)