File tree Expand file tree Collapse file tree 1 file changed +1
-7
lines changed Expand file tree Collapse file tree 1 file changed +1
-7
lines changed Original file line number Diff line number Diff line change @@ -166,12 +166,7 @@ def is_ht_enabled():
166
166
hostname = socket .gethostname ()
167
167
168
168
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 )
175
170
176
171
# get parameters that are common for all cases
177
172
common_params = config ['common' ]
@@ -267,7 +262,6 @@ class GenerationArgs:
267
262
env = os .environ .copy ()
268
263
if lib == 'xgboost' :
269
264
env ['OMP_NUM_THREADS' ] = omp_num_threads
270
- env ['OMP_PLACES' ] = omp_places
271
265
272
266
for i , case in enumerate (cases ):
273
267
command = f'python { lib } /{ algorithm } .py --batch { batch } ' \
You can’t perform that action at this time.
0 commit comments