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 50997035be [ci] Restructure Jenkinsfile (#11380)
add bbc6ba362f [Meta Schedule] Add Auto-Thread Binding Rule (#11177)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/mutator.h | 10 +-
include/tvm/meta_schedule/postproc.h | 4 +-
include/tvm/meta_schedule/schedule_rule.h | 7 +
python/tvm/meta_schedule/mutator/__init__.py | 1 +
.../{mutate_unroll.py => mutate_thread_binding.py} | 11 +-
.../postproc/rewrite_unbound_block.py | 5 +-
python/tvm/meta_schedule/schedule_rule/__init__.py | 1 +
.../{cross_thread_reduction.py => auto_bind.py} | 28 +--
.../meta_schedule/testing/conv2d_winograd_cpu.py | 2 +-
.../meta_schedule/testing/conv2d_winograd_cuda.py | 2 +-
python/tvm/meta_schedule/testing/schedule_rule.py | 8 +
python/tvm/meta_schedule/tune.py | 8 +-
python/tvm/topi/cuda/conv2d_nhwc_winograd.py | 2 +-
python/tvm/topi/cuda/conv2d_winograd.py | 2 +-
python/tvm/topi/nn/conv2d.py | 7 +-
src/meta_schedule/mutator/mutate_thread_binding.cc | 167 ++++++++++++++++++
.../postproc/rewrite_unbound_block.cc | 139 ++-------------
src/meta_schedule/schedule_rule/auto_bind.cc | 192 +++++++++++++++++++++
src/meta_schedule/schedule_rule/auto_bind.h | 52 ++++++
src/meta_schedule/schedule_rule/winograd.cc | 23 ++-
...test_meta_schedule_custom_rule_winograd_cuda.py | 96 ++++++++++-
..._meta_schedule_mutator_mutate_thread_binding.py | 86 +++++++++
... test_meta_schedule_schedule_rule_auto_bind.py} | 55 +++---
23 files changed, 724 insertions(+), 184 deletions(-)
copy python/tvm/meta_schedule/mutator/{mutate_unroll.py =>
mutate_thread_binding.py} (76%)
copy python/tvm/meta_schedule/schedule_rule/{cross_thread_reduction.py =>
auto_bind.py} (57%)
create mode 100644 src/meta_schedule/mutator/mutate_thread_binding.cc
create mode 100644 src/meta_schedule/schedule_rule/auto_bind.cc
create mode 100644 src/meta_schedule/schedule_rule/auto_bind.h
create mode 100644
tests/python/unittest/test_meta_schedule_mutator_mutate_thread_binding.py
copy tests/python/unittest/{test_meta_schedule_schedule_rule_add_rfactor.py =>
test_meta_schedule_schedule_rule_auto_bind.py} (57%)
