Skip to content

Commit f6e0c38

Browse files
committed
fix for log leveling inside docker
1 parent f2daf74 commit f6e0c38

File tree

2 files changed

+28
-4
lines changed

2 files changed

+28
-4
lines changed

automation/script/docker.py

+25-1
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,32 @@ def dockerfile(self_module, input_params):
178178
'comments': comments, 'run_cmd': f"{run_command_string} --quiet",
179179
'script_tags': input_params.get('tags'), 'env': env,
180180
'dockerfile_env': dockerfile_env,
181-
'quiet': True, 'real_run': True, '-v': is_true(input_params.get('v', input_params.get('verbose', False))), '-s': is_true(input_params.get('s', input_params.get('silent', False)))
181+
'quiet': True, 'real_run': True
182182
}
183+
184+
docker_v = False
185+
docker_s = False
186+
if is_true(input_params.get('docker_v', input_params.get('docker_verbose', False))):
187+
docker_v = True
188+
if is_true(input_params.get('docker_s', input_params.get('docker_silent', False))):
189+
docker_s = True
190+
191+
if docker_s and docker_v:
192+
logger.warning(
193+
"Both verbose and silent is set to True. Verbose will take precedence.")
194+
docker_s = False
195+
196+
if not docker_s and not docker_v:
197+
if logger.level == logging.DEBUG:
198+
docker_v = True
199+
elif logger.level == logging.WARNING:
200+
docker_s = True
201+
202+
if docker_s:
203+
mlc_docker_input['run_cmd'] += ' -s'
204+
elif docker_v:
205+
mlc_docker_input['run_cmd'] += ' -v'
206+
183207
mlc_docker_input.update(docker_inputs)
184208

185209
dockerfile_result = self_module.action_object.access(mlc_docker_input)

script/run-mlperf-inference-app/customize.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def preprocess(i):
159159
tags = "app,mlperf,inference,generic," + variation_implementation + variation_model + variation_backend + variation_device + \
160160
variation_run_style + variation_reproducibility + \
161161
variation_quantization_string + power_variation + variation_all_models
162-
verbose = inp.get('v', False)
162+
163163
print_env = inp.get('print_env', False)
164164
print_deps = inp.get('print_deps', False)
165165
add_deps_recursive = inp.get('add_deps_recursive', {})
@@ -273,7 +273,7 @@ def preprocess(i):
273273
print(f"\nRunning loadgen scenario: {scenario} and mode: {mode}")
274274
ii = {'action': action, 'automation': 'script', 'tags': scenario_tags, 'quiet': 'true',
275275
'env': env_copy, 'const': const_copy, 'input': inp, 'state': state, 'add_deps': copy.deepcopy(add_deps), 'add_deps_recursive':
276-
copy.deepcopy(add_deps_recursive), 'ad': ad, 'adr': copy.deepcopy(adr), 'v': verbose, 'print_env': print_env, 'print_deps': print_deps, 'dump_version_info': dump_version_info}
276+
copy.deepcopy(add_deps_recursive), 'ad': ad, 'adr': copy.deepcopy(adr), 'print_env': print_env, 'print_deps': print_deps, 'dump_version_info': dump_version_info}
277277

278278
if action == "docker":
279279
for k in docker_extra_input:
@@ -311,7 +311,7 @@ def preprocess(i):
311311
env['MLC_MLPERF_LOADGEN_MODE'] = "compliance"
312312
ii = {'action': action, 'automation': 'script', 'tags': scenario_tags, 'quiet': 'true',
313313
'env': copy.deepcopy(env), 'const': copy.deepcopy(const), 'input': inp, 'state': state, 'add_deps': copy.deepcopy(add_deps), 'add_deps_recursive':
314-
copy.deepcopy(add_deps_recursive), 'adr': copy.deepcopy(adr), 'ad': ad, 'v': verbose, 'print_env': print_env, 'print_deps': print_deps, 'dump_version_info': dump_version_info}
314+
copy.deepcopy(add_deps_recursive), 'adr': copy.deepcopy(adr), 'ad': ad, 'print_env': print_env, 'print_deps': print_deps, 'dump_version_info': dump_version_info}
315315
if action == "docker":
316316
for k in docker_extra_input:
317317
ii[k] = docker_extra_input[k]

0 commit comments

Comments
 (0)