Skip to content

Commit

Permalink
Fix _short tag support
Browse files Browse the repository at this point in the history
  • Loading branch information
arjunsuresh committed Sep 19, 2024
1 parent 3e24bb9 commit 8d6392d
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -419,9 +419,14 @@ def mlperf_inference_run_command(spaces, model, implementation, framework, categ
if "dlrm-v2" in model.lower() and implementation == "nvidia":
docker_cmd_suffix += f" \\\n{pre_space} --criteo_day23_raw_data_path=<PATH_TO_CRITEO_DAY23_RAW_DATA>"

if "short" in extra_variation_tags:
full_ds_needed_tag = ""
else:
full_ds_needed_tag = ",_full"

docker_setup_cmd = f"""\n
{f_pre_space}```bash
{f_pre_space}cm run script --tags=run-mlperf,inference,_find-performance,_full,_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
{f_pre_space}cm run script --tags=run-mlperf,inference,_find-performance,{full_ds_needed_tag}_{code_version}{scenario_variation_tag}{extra_variation_tags} \\
{pre_space} --model={model} \\
{pre_space} --implementation={implementation} \\
{pre_space} --framework={framework} \\
Expand Down

0 comments on commit 8d6392d

Please sign in to comment.