Added more info to metadata

hpvm/projects/torch2hpvm/torch2hpvm/approxknobs.json

@@ -2,7 +2,8 @@
     {
         "name": "11",
         "speedup": 1.0,
-        "applies_to": null
+        "applies_to": null,
+        "is_baseline": true
     },
     {
         "name": "12",

hpvm/projects/torch2hpvm/torch2hpvm/compile.py

@@ -98,14 +98,21 @@ class ModelExporter:
         KnobInfoT = Tuple[str, float]
         ty_knobs: Dict[str, List[KnobInfoT]] = defaultdict(list)
         default_knobs: List[KnobInfoT] = []
+        baseline_knob = None
         for k in knobs:
-            applies_to = k.pop("applies_to")
-            k = k["name"], k["speedup"]
+            kp = k["name"], k["speedup"]
+            if "is_baseline" in k:
+                if baseline_knob:
+                    raise ValueError("Multiple baseline knobs")
+                baseline_knob = k["name"]
+            applies_to = k["applies_to"]
             if applies_to is None:
-                default_knobs.append(k)
+                default_knobs.append(kp)
                 continue
             for ty in applies_to:
-                ty_knobs[ty].append(k)
+                ty_knobs[ty].append(kp)
+        if not baseline_knob:
+            raise ValueError("No baseline knob given")
         idx = 0
         op_cost: Dict[str, int] = {}
         op_knobs: Dict[str, List[str]] = {}
@@ -127,6 +134,9 @@ class ModelExporter:
                     "op_cost": op_cost,
                     "knob_speedup": knob_speedup,
                     "op_knobs": op_knobs,
+                    "baseline_knob": baseline_knob,
+                    "tune_args": "tune",
+                    "test_args": "test"
                 },
                 f,
                 indent=2,