Output the submission directory in push to GitHub

script/push-mlperf-inference-results-to-github/customize.py

@@ -42,3 +42,15 @@ def preprocess(i):
         env['MLC_GIT_PUSH_CMD'] = "git push"
 
     return {'return': 0}
+
+
+def postprocess(i):
+
+    os_info = i['os_info']
+    env = i['env']
+    meta = i['meta']
+    automation = i['automation']
+
+    env['MLC_MLPERF_RESULTS_SYNC_DIR'] = env['MLC_GIT_REPO_CHECKOUT_PATH']
+
+    return {'return': 0}

script/push-mlperf-inference-results-to-github/meta.yaml

@@ -4,6 +4,10 @@ automation_uid: 5b4e0237da074764
 category: MLPerf benchmark support
 default_env:
   MLC_MLPERF_RESULTS_GIT_REPO_URL: https://github.com/mlcommons/mlperf_inference_submissions_v4.0
+new_env_keys: 
+  - MLC_MLPERF_RESULTS_SYNC_DIR
+print_env_at_the_end:
+  MLC_MLPERF_RESULTS_SYNC_DIR: Path to the locally synced submission directory
 deps:
 - names:
   - python3