From 50d251406b18ccdb53524c441fce258a07d9f263 Mon Sep 17 00:00:00 2001 From: Arjun Suresh Date: Fri, 15 Mar 2024 18:28:34 +0530 Subject: [PATCH] Support sut variation in run-mlperf-inference-app --- cm-mlops/script/run-mlperf-inference-app/_cm.yaml | 1 + cm-mlops/script/run-mlperf-inference-app/customize.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/cm-mlops/script/run-mlperf-inference-app/_cm.yaml b/cm-mlops/script/run-mlperf-inference-app/_cm.yaml index 3119a3431..09ba3dc78 100644 --- a/cm-mlops/script/run-mlperf-inference-app/_cm.yaml +++ b/cm-mlops/script/run-mlperf-inference-app/_cm.yaml @@ -90,6 +90,7 @@ input_mapping: test_query_count: CM_TEST_QUERY_COUNT threads: CM_NUM_THREADS batch_size: CM_MLPERF_LOADGEN_MAX_BATCHSIZE + sut: CM_MLPERF_INFERENCE_SUT_VARIATION new_state_keys: - app_mlperf_inference_* diff --git a/cm-mlops/script/run-mlperf-inference-app/customize.py b/cm-mlops/script/run-mlperf-inference-app/customize.py index 0872c7640..9d634b29d 100644 --- a/cm-mlops/script/run-mlperf-inference-app/customize.py +++ b/cm-mlops/script/run-mlperf-inference-app/customize.py @@ -134,6 +134,11 @@ def preprocess(i): add_deps_recursive['mlperf-inference-implementation'] = {} add_deps_recursive['mlperf-inference-implementation']['tags'] = "_batch_size."+env['CM_MLPERF_LOADGEN_MAX_BATCHSIZE'] + if env.get('CM_MLPERF_SUT_VARIATION', '') != '': + if not add_deps_recursive.get('mlperf-inference-implementation', {}): + add_deps_recursive['mlperf-inference-implementation'] = {} + add_deps_recursive['mlperf-inference-implementation']['tags'] = "_"+env['CM_MLPERF_SUT_VARIATION'] + if env.get('CM_NETWORK_LOADGEN', '') != '': if not add_deps_recursive.get('mlperf-inference-implementation', {}): add_deps_recursive['mlperf-inference-implementation'] = {}