Export VERSION for mlperf inference src

mlcommons · Oct 21, 2024 · 2ba815e · 2ba815e
1 parent 4e954c3
commit 2ba815e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/script/get-mlperf-inference-src/_cm.json b/script/get-mlperf-inference-src/_cm.json
@@ -32,6 +32,7 @@
     "CM_MLPERF_INFERENCE_GPTJ_PATH",
     "CM_MLPERF_INFERENCE_RNNT_PATH",
     "CM_MLPERF_INFERENCE_SOURCE",
+    "CM_MLPERF_INFERENCE_SOURCE_VERSION",
     "CM_MLPERF_INFERENCE_VERSION",
     "CM_MLPERF_INFERENCE_VISION_PATH",
     "CM_MLPERF_LAST_RELEASE",

diff --git a/script/get-mlperf-inference-src/customize.py b/script/get-mlperf-inference-src/customize.py
@@ -101,6 +101,11 @@ def postprocess(i):
     env['+PYTHONPATH']=[]
     env['+PYTHONPATH'].append(os.path.join(env['CM_MLPERF_INFERENCE_CLASSIFICATION_AND_DETECTION_PATH'], 'python'))
 
+    if os.path.exists(os.path.join(inference_root, "loadgen", "VERSION")):
+        with open(os.path.join(inference_root, "loadgen", "VERSION") as f:
+            version_info = f.read()
+        env['CM_MLPERF_INFERENCE_SOURCE_VERSION'] = version_info
+
     if env.get('CM_GET_MLPERF_IMPLEMENTATION_ONLY', '') == "yes":
         return {'return':0}