You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by sy...@apache.org on 2021/12/19 08:08:49 UTC
[tvm] branch main updated (115919b -> 9ee0007)
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 115919b [MetaSchedule] Misc improvement of the Measurer (#9757)
add 9ee0007 [M3c][MetaScheduler] Add ScheduleRule class & PostOrderApply space generator. (#9761)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/schedule_rule.h | 195 ++++++++++++
include/tvm/meta_schedule/search_strategy.h | 6 +-
include/tvm/meta_schedule/space_generator.h | 16 +-
include/tvm/meta_schedule/tune_context.h | 5 +
include/tvm/tir/schedule/schedule.h | 6 +
python/tvm/meta_schedule/__init__.py | 1 +
.../schedule_rule}/__init__.py | 10 +-
.../meta_schedule/schedule_rule/schedule_rule.py | 96 ++++++
.../tvm/meta_schedule/space_generator/__init__.py | 2 +-
...pace_generator_union.py => post_order_apply.py} | 29 +-
.../space_generator/space_generator.py | 5 +-
python/tvm/meta_schedule/tune_context.py | 27 +-
src/meta_schedule/schedule_rule/schedule_rule.cc | 55 ++++
src/meta_schedule/search_strategy/replay_trace.cc | 8 +-
.../space_generator/post_order_apply.cc | 155 ++++++++++
.../space_generator/space_generator_union.cc | 4 +-
src/meta_schedule/tune_context.cc | 7 +-
src/meta_schedule/utils.h | 4 +
src/tir/schedule/concrete_schedule.h | 14 +
.../test_meta_schedule_post_order_apply.py | 342 +++++++++++++++++++++
20 files changed, 924 insertions(+), 63 deletions(-)
create mode 100644 include/tvm/meta_schedule/schedule_rule.h
copy python/tvm/{_ffi/_cy2 => meta_schedule/schedule_rule}/__init__.py (73%)
create mode 100644 python/tvm/meta_schedule/schedule_rule/schedule_rule.py
copy python/tvm/meta_schedule/space_generator/{space_generator_union.py => post_order_apply.py} (61%)
create mode 100644 src/meta_schedule/schedule_rule/schedule_rule.cc
create mode 100644 src/meta_schedule/space_generator/post_order_apply.cc
create mode 100644 tests/python/unittest/test_meta_schedule_post_order_apply.py