Merge branch 'dev' into dev

mlcommons · Feb 5, 2025 · d5b6e87 · d5b6e87
2 parents 0165cbc + 0f838b3
commit d5b6e87
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/script/get-dataset-igbh/customize.py b/script/get-dataset-igbh/customize.py
@@ -43,7 +43,7 @@ def preprocess(i):
     run_cmd += x_sep + \
         f"""{
             env['MLC_PYTHON_BIN_WITH_PATH']} tools/split_seeds.py --path {download_loc} --dataset_size {
-            env['MLC_DATASET_IGBH_SIZE']} """
+            env['MLC_DATASET_IGBH_SIZE']} {env.get('MLC_IGBH_CALIBRATION_FLAG', '')} """
 
     # compress graph(for glt implementation)
     if env.get('MLC_IGBH_GRAPH_COMPRESS', '') == "yes":

diff --git a/script/get-dataset-igbh/meta.yaml b/script/get-dataset-igbh/meta.yaml
@@ -437,6 +437,9 @@ variations:
   glt:
     env:
       MLC_IGBH_GRAPH_COMPRESS: yes
+  calibration:
+    env:
+      MLC_IGBH_CALIBRATION_FLAG: --calibration
   csc:
     group: compressed-layout
     default: true