This is an automated email from the ASF dual-hosted git repository.
xiyou pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from e1ac5658ad [Hexagon]Pull and build specific LLVM sha for hexagon
docker CI (#13199)
add f42826eec4 [MetaSchedule] Enable anchor-block tuning (#13206)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/database.h | 8 +
include/tvm/meta_schedule/schedule_rule.h | 5 +-
python/tvm/meta_schedule/__init__.py | 1 +
python/tvm/meta_schedule/database/json_database.py | 4 +
.../tvm/meta_schedule/database/memory_database.py | 4 +
.../meta_schedule/database/schedule_fn_database.py | 4 +
python/tvm/meta_schedule/relay_integration.py | 8 +
python/tvm/meta_schedule/trace_apply.py | 39 +
python/tvm/meta_schedule/tune.py | 4 +
python/tvm/script/tir/__init__.py | 5 +-
python/tvm/script/tir/ty.py | 2 +-
python/tvm/tir/schedule/analysis.py | 18 +
src/meta_schedule/module_equality.cc | 24 +
src/meta_schedule/module_equality.h | 4 +
src/meta_schedule/schedule_rule/auto_bind.cc | 5 +-
src/meta_schedule/schedule_rule/schedule_rule.cc | 18 +-
.../space_generator/space_generator.cc | 7 +-
src/meta_schedule/trace_apply.cc | 235 ++
src/meta_schedule/trace_apply.h | 48 +
src/meta_schedule/utils.h | 36 +
src/relay/backend/task_extraction.cc | 75 +-
src/relay/backend/te_compiler_cache.cc | 20 +-
src/target/target_kind.cc | 3 +
src/tir/schedule/analysis/analysis.cc | 2 +
src/tir/schedule/primitive/cache_read_write.cc | 7 +-
src/tir/schedule/utils.h | 46 +
.../metaschedule_e2e/test_resnet50_int8.py | 8 +-
.../test_meta_schedule_relay_integration.py | 145 ++
.../unittest/test_meta_schedule_trace_apply.py | 2745 ++++++++++++++++++++
.../test_meta_schedule_vnni_integration.py | 2 +
30 files changed, 3486 insertions(+), 46 deletions(-)
create mode 100644 python/tvm/meta_schedule/trace_apply.py
create mode 100644 src/meta_schedule/trace_apply.cc
create mode 100644 src/meta_schedule/trace_apply.h
create mode 100644 tests/python/unittest/test_meta_schedule_trace_apply.py
