Merge from CTuning (#1011)

mlcommons · Nov 12, 2023 · d207aae · d207aae
2 parents 5d3bb34 + 0173b9e
commit d207aae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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'] = ""