From 14c95123486e31c7cec233287c7093c5def9e244 Mon Sep 17 00:00:00 2001
From: Arjun <arjunsuresh1987@gmail.com>
Date: Fri, 21 Feb 2025 01:31:40 +0000
Subject: [PATCH 1/3] Support docker script for mobilenet models

---
 .../meta.yaml                                 | 36 ++++++++++++++-----
 1 file changed, 28 insertions(+), 8 deletions(-)

diff --git a/script/run-mlperf-inference-mobilenet-models/meta.yaml b/script/run-mlperf-inference-mobilenet-models/meta.yaml
index fd4cf3468..93d8ecc17 100644
--- a/script/run-mlperf-inference-mobilenet-models/meta.yaml
+++ b/script/run-mlperf-inference-mobilenet-models/meta.yaml
@@ -8,22 +8,42 @@ default_env:
   MLC_MLPERF_RUN_FP32: 'yes'
   MLC_MLPERF_RUN_INT8: 'yes'
   MLC_MLPERF_RUN_MOBILENETS: 'no'
+  MLC_USE_DATASET_FROM_HOST: 'yes'
 deps:
 - tags: get,sys-utils-mlc
 docker:
+  deps:
+    - tags: get,mlperf,inference,results,dir,local
+      names:
+      - get-mlperf-inference-results-dir
+      skip_if_env:
+        OUTPUT_BASE_DIR: [ on ]
+    - tags: get,mlperf,inference,submission,dir,local
+      names:
+      - get-mlperf-inference-submission-dir
+      skip_if_env:
+        MLC_MLPERF_INFERENCE_SUBMISSION_DIR: [ on ]
+    - tags: get,dataset,imagenet,validation,original,_full
+      enable_if_env:
+        MLC_USE_DATASET_FROM_HOST:
+        - 'yes'
+      names:
+        - imagenet-original
+        - dataset-original
   input_mapping:
-    imagenet_path: IMAGENET_PATH
-    results_dir: RESULTS_DIR
-    submission_dir: SUBMISSION_DIR
+    imagenet_path: MLC_DATASET_IMAGENET_PATH
+    results_dir: MLC_MLPERF_INFERENCE_RESULTS_DIR
+    submission_dir: MLC_MLPERF_INFERENCE_SUBMISSION_DIR
   docker_run_final_cmds:
-  - mlcr run,mlperf,inference,mobilenet-models,_find-performance --adr.mlperf-inference-implementation.fake_run=True
-    --adr.compiler.tags=gcc
+  - mlcr run,mlperf,inference,mobilenet-models,_find-performance --adr.mlperf-inference-implementation.fake_run=True --adr.compiler.tags=gcc
   fake_run_deps: false
   mounts:
-  - ${{ IMAGENET_PATH }}:${{ IMAGENET_PATH }}
-  - ${{ RESULTS_DIR }}:${{ RESULTS_DIR }}
-  - ${{ SUBMISSION_DIR }}:${{ SUBMISSION_DIR }}
+   - "${{ MLC_DATASET_IMAGENET_PATH }}:${{ MLC_DATASET_IMAGENET_PATH }}"
+   - "${{ MLC_MLPERF_INFERENCE_RESULTS_DIR }}:${{ MLC_MLPERF_INFERENCE_RESULTS_DIR }}"
+   - "${{ OUTPUT_BASE_DIR }}:${{ OUTPUT_BASE_DIR }}"
+   - "${{ MLC_MLPERF_INFERENCE_SUBMISSION_DIR }}:${{ MLC_MLPERF_INFERENCE_SUBMISSION_DIR }}"
   run: true
+  docker_it: true
 input_mapping:
   find-performance: MLC_MLPERF_FIND_PERFORMANCE_MODE
   imagenet_path: IMAGENET_PATH

From c516b2c1d0804277ee81519a3feef43bd80cc4e2 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjunsuresh1987@gmail.com>
Date: Fri, 21 Feb 2025 16:19:43 +0000
Subject: [PATCH 2/3] Added torch deps for igbh

---
 script/get-dataset-igbh/meta.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/script/get-dataset-igbh/meta.yaml b/script/get-dataset-igbh/meta.yaml
index b48e7b42c..4fa775724 100644
--- a/script/get-dataset-igbh/meta.yaml
+++ b/script/get-dataset-igbh/meta.yaml
@@ -24,9 +24,10 @@ deps:
   - tags: get,python
     names:
       - get-python
-  - tags: get,generic-python-lib,_package.igb,_url.git+https://github.com/gateoverflow/IGB-Datasets.git
+  - tags: get,generic-python-lib,_package.igb
   - tags: get,generic-python-lib,_package.colorama
   - tags: get,generic-python-lib,_package.tqdm
+  - tags: get,generic-python-lib,_package.torch
 
 
 prehook_deps:

From fceaaba2491e98166d5f9a68cdd3f62151846c0e Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjun@gateoverflow.com>
Date: Fri, 21 Feb 2025 21:49:47 +0000
Subject: [PATCH 3/3] Update meta.yaml

---
 script/get-dataset-igbh/meta.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/script/get-dataset-igbh/meta.yaml b/script/get-dataset-igbh/meta.yaml
index 4fa775724..2f6dc96ad 100644
--- a/script/get-dataset-igbh/meta.yaml
+++ b/script/get-dataset-igbh/meta.yaml
@@ -24,7 +24,7 @@ deps:
   - tags: get,python
     names:
       - get-python
-  - tags: get,generic-python-lib,_package.igb
+  - tags: get,generic-python-lib,_package.igb,_url.git+https://github.com/IllinoisGraphBenchmark/IGB-Datasets
   - tags: get,generic-python-lib,_package.colorama
   - tags: get,generic-python-lib,_package.tqdm
   - tags: get,generic-python-lib,_package.torch