From 47825fe33b3963a6a6a2a3aec4d6699bc015b286 Mon Sep 17 00:00:00 2001 From: Arjun Suresh Date: Sun, 12 Nov 2023 20:46:10 +0530 Subject: [PATCH] Support multi-gpu runs for bert --- cm-mlops/script/app-mlperf-inference-reference/_cm.yaml | 2 ++ cm-mlops/script/app-mlperf-inference-reference/customize.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml b/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml index 91c8121e05..d7cdf23e0d 100644 --- a/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml +++ b/cm-mlops/script/app-mlperf-inference-reference/_cm.yaml @@ -473,6 +473,8 @@ deps: names: - mlperf-implementation + - tags: get,generic-python-lib,_package.psutil + posthook_deps: - names: - mlperf-runner diff --git a/cm-mlops/script/app-mlperf-inference-reference/customize.py b/cm-mlops/script/app-mlperf-inference-reference/customize.py index 6c0cc8e0e5..d179caf124 100644 --- a/cm-mlops/script/app-mlperf-inference-reference/customize.py +++ b/cm-mlops/script/app-mlperf-inference-reference/customize.py @@ -38,7 +38,7 @@ def preprocess(i): # "val_map.txt") # ret = os.system(cmd) - env['CM_MLPERF_LOADGEN_EXTRA_OPTIONS'] = "" + env['CM_MLPERF_LOADGEN_EXTRA_OPTIONS'] = " " + env.get('CM_MLPERF_LOADGEN_EXTRA_OPTIONS', '') + " " if 'CM_MLPERF_LOADGEN_QPS' not in env: env['CM_MLPERF_LOADGEN_QPS_OPT'] = ""