Skip to content

Commit 9ee25eb

Browse files
authored
[MetaSchedule] Added a cost model (#11961)
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.
1 parent 2d50728 commit 9ee25eb

File tree

3 files changed

+1012
-2
lines changed

3 files changed

+1012
-2
lines changed

python/tvm/meta_schedule/cost_model/cost_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def update(
190190
raise NotImplementedError
191191

192192
def predict(self, context: TuneContext, candidates: List[MeasureCandidate]) -> np.ndarray:
193-
"""Update the cost model given running results.
193+
"""Predict given the measure candidates.
194194
195195
Parameters
196196
----------

0 commit comments

Comments
 (0)