diff --git a/script/app-mlperf-inference-mlcommons-python/_cm.yaml b/script/app-mlperf-inference-mlcommons-python/_cm.yaml index 45401431f..1c1a89eee 100644 --- a/script/app-mlperf-inference-mlcommons-python/_cm.yaml +++ b/script/app-mlperf-inference-mlcommons-python/_cm.yaml @@ -625,9 +625,6 @@ deps: enable_if_env: CM_MODEL: - rgat - skip_if_any_env: - CM_DATASET_IGBH_PATH: - - "on" skip_if_env: CM_RUN_STATE_DOCKER: - 'yes' diff --git a/script/build-dockerfile/customize.py b/script/build-dockerfile/customize.py index baf0860f8..8f451ebb6 100644 --- a/script/build-dockerfile/customize.py +++ b/script/build-dockerfile/customize.py @@ -379,6 +379,26 @@ def preprocess(i): if run_cmd_extra != '': x += ' ' + run_cmd_extra + if env.get('CM_DOCKER_RUN_SCRIPT_TAGS', '') != '' and str(env.get( + 'CM_DOCKER_ADD_DEPENDENT_SCRIPTS_RUN_COMMANDS', '')).lower() in ["yes", "1", "true"]: + cm_input = {'action': 'run', + 'automation': 'script', + 'tags': f"""{env['CM_DOCKER_RUN_SCRIPT_TAGS']}""", + 'print_deps': True, + 'quiet': True, + 'silent': True, + 'fake_run': True, + 'fake_deps': True + } + r = self_module.cmind.access(cm_input) + if r['return'] > 0: + return r + print_deps = r['new_state']['print_deps'] + fake_run_str = " --fake_run" if env.get('CM_DOCKER_FAKE_DEPS') else "" + cmds = ["RUN " + dep for dep in print_deps] + for cmd in cmds: + f.write(cmd + fake_run_str + EOL) + f.write(x + EOL) # fake_run to install the dependent scripts and caching them