[MetaSchedule] Added a cost model (apache#11961)

In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
masahi · Jul 15, 2022 · 77d2ee2 · 77d2ee2
1 parent ca14468
commit 77d2ee2
Show file tree

Hide file tree

Showing 3 changed files with 1,012 additions and 2 deletions.
diff --git a/python/tvm/meta_schedule/cost_model/cost_model.py b/python/tvm/meta_schedule/cost_model/cost_model.py
@@ -190,7 +190,7 @@ def update(
         raise NotImplementedError
 
     def predict(self, context: TuneContext, candidates: List[MeasureCandidate]) -> np.ndarray:
-        """Update the cost model given running results.
+        """Predict given the measure candidates.
 
         Parameters
         ----------