Skip to content

Commit bd838e8

Browse files
committed
for debug
Signed-off-by: chensuyue <[email protected]>
1 parent 5da0d8b commit bd838e8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

neural_compressor/experimental/benchmark.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ def __call__(self, mode='performance'):
189189
if os.environ.get('NC_ENV_CONF') == 'True':
190190
return self.run_instance(mode)
191191
else:
192+
print("---------start to config instance--->")
192193
self.config_instance()
193194
self.summary_benchmark()
194195
return None
@@ -221,6 +222,7 @@ def summary_benchmark(self):
221222

222223
def config_instance(self):
223224
"""Configure the multi-instance commands and trigger benchmark with sub process."""
225+
print("---------config_instance--->")
224226
raw_cmd = sys.executable + ' ' + ' '.join(sys.argv)
225227
multi_instance_cmd = ''
226228
num_of_instance = int(os.environ.get('NUM_OF_INSTANCE'))
@@ -249,7 +251,7 @@ def config_instance(self):
249251
multi_instance_cmd += '{} \n'.format(instance_cmd)
250252

251253
multi_instance_cmd += 'wait' if sys.platform in ['linux'] else ''
252-
logger.info("Running command is\n{}".format(multi_instance_cmd))
254+
print("Running command is\n{}".format(multi_instance_cmd))
253255
# each instance will execute single instance
254256
set_env_var('NC_ENV_CONF', True, overwrite_existing=True)
255257
if sys.platform in ['linux']:
@@ -281,6 +283,7 @@ def generate_prefix(self, core_list):
281283
if cores_per_socket % cores_per_instance == 0:
282284
from functools import reduce
283285
hex_core = hex(reduce(lambda x, y : x | y, [1 << p for p in core_list]))
286+
print("return prefix for windows --> ")
284287
return 'start /b /WAIT /node {} /affinity {} CMD /c'.format(socket_id, hex_core)
285288
else:
286289
return ''

0 commit comments

Comments
 (0)