Preprocess mlperf inference submission, code cleanup

GATEOverflow · Nov 7, 2024 · 21425ef · 21425ef
1 parent 69daefe
commit 21425ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/script/preprocess-mlperf-inference-submission/_cm.json b/script/preprocess-mlperf-inference-submission/_cm.json
@@ -31,6 +31,7 @@
     }
   ],
   "input_mapping": {
+    "input": "CM_MLPERF_INFERENCE_SUBMISSION_DIR",
     "submission_dir": "CM_MLPERF_INFERENCE_SUBMISSION_DIR",
     "submitter": "CM_MLPERF_SUBMITTER"
   },

diff --git a/script/preprocess-mlperf-inference-submission/customize.py b/script/preprocess-mlperf-inference-submission/customize.py
@@ -14,7 +14,7 @@ def preprocess(i):
         print("Please set CM_MLPERF_INFERENCE_SUBMISSION_DIR")
         return {'return': 1, 'error':'CM_MLPERF_INFERENCE_SUBMISSION_DIR is not specified'}
 
-    submitter = env.get("CM_MLPERF_SUBMITTER", "cTuning")
+    submitter = env.get("CM_MLPERF_SUBMITTER", "MLCommons")
     submission_processed = submission_dir + "_processed"
 
     if os.path.exists(submission_processed):