This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from c2488ac [MetaSchedule] Extract task weights during task extraction
(#10810)
add ce28068 [MetaSchedule] Support grouping in the cost model (#10811)
No new revisions were added by this update.
Summary of changes:
python/tvm/meta_schedule/cost_model/xgb_model.py | 243 +++++++++++++--------
.../testing/tune_relay_auto_scheduler.py | 30 ++-
.../testing/tune_relay_meta_schedule.py | 87 +++++++-
python/tvm/meta_schedule/tune.py | 13 +-
python/tvm/meta_schedule/utils.py | 11 +-
python/tvm/relay/build_module.py | 2 +-
python/tvm/rpc/client.py | 10 +-
src/meta_schedule/tune_context.cc | 3 +
src/meta_schedule/utils.h | 15 ++
.../unittest/test_meta_schedule_cost_model.py | 50 +++--
10 files changed, 325 insertions(+), 139 deletions(-)
