Skip to content

Commit 9825d35

Browse files
authored
Fix SUT name update in mlperf-inference-submission-generation (#96)
* Remove ulimit memlock option for nvidia-mlperf-inference * Added an option to override SUT name during submission generation, turn on checker by default
1 parent d9551f8 commit 9825d35

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

script/generate-mlperf-inference-submission/_cm.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,9 @@ post_deps:
101101
- 'yes'
102102
- true
103103
tags: preprocess,mlperf,submission
104-
- enable_if_env:
104+
- skip_if_env:
105105
CM_RUN_SUBMISSION_CHECKER:
106-
- 'yes'
106+
- 'no'
107107
names:
108108
- mlperf-inference-submission-checker
109109
- submission-checker

script/generate-mlperf-inference-submission/customize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def generate_submission(env, state, inp, submission_division):
190190
result_path, 'system_meta.json')
191191
# checks for json file containing system meta
192192
sut_info = {
193-
"hardware_name": None,
193+
"system_name": None,
194194
"implementation": None,
195195
"device": None,
196196
"framework": None,
@@ -276,7 +276,7 @@ def generate_submission(env, state, inp, submission_division):
276276
{model: returned_model_name})
277277

278278
if check_dict_filled(sut_info.keys(), sut_info):
279-
system = sut_info["hardware_name"]
279+
system = env.get('CM_HW_NAME', sut_info["system_name"])
280280
implementation = sut_info["implementation"]
281281
device = sut_info["device"]
282282
framework = sut_info["framework"].replace(" ", "_")

0 commit comments

Comments
 (0)