9ee25eb9f4
In this PR, I added a cost model based on SegmentSum MLP, which can be used for pre-training or integration with TVM.