This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 81b66e6 [Docker] Relax name check (#10011)
add 1ac01b4 [MetaSchedule] Schedule Rule: Cross Thread Reduction (#9994)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/schedule_rule.h | 7 +
python/tvm/meta_schedule/schedule_rule/__init__.py | 1 +
...mpute_location.py => cross_thread_reduction.py} | 22 +-
python/tvm/meta_schedule/testing/schedule_rule.py | 8 +
.../schedule_rule/cross_thread_reduction.cc | 285 +++++++++++++++++++++
src/tir/schedule/analysis/analysis.cc | 5 +-
src/tir/schedule/trace.cc | 6 +-
src/tir/schedule/utils.h | 15 ++
...chedule_schedule_rule_cross_thread_reduction.py | 241 +++++++++++++++++
9 files changed, 580 insertions(+), 10 deletions(-)
copy python/tvm/meta_schedule/schedule_rule/{random_compute_location.py =>
cross_thread_reduction.py} (58%)
create mode 100644 src/meta_schedule/schedule_rule/cross_thread_reduction.cc
create mode 100644
tests/python/unittest/test_meta_schedule_schedule_rule_cross_thread_reduction.py
