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 ec24ae60a0 [BYOC] RelayToTIR custom codegen passes can still depend on
dynamic shape functions (#11619)
add 6fca5c657a [MetaSchedule] Developer Ergonomics Enhancement (#11622)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/search_strategy.h | 11 +-
include/tvm/meta_schedule/tune_context.h | 27 +-
python/tvm/meta_schedule/__init__.py | 5 +-
python/tvm/meta_schedule/database/__init__.py | 1 +
python/tvm/meta_schedule/database/database.py | 2 +-
.../tvm/meta_schedule/database/memory_database.py | 63 ++++
python/tvm/meta_schedule/default_config.py | 346 +++++++++++++++++++++
python/tvm/meta_schedule/mutator/mutator.py | 8 +-
python/tvm/meta_schedule/postproc/postproc.py | 6 +-
.../meta_schedule/schedule_rule/schedule_rule.py | 10 +-
.../search_strategy/search_strategy.py | 13 +-
.../meta_schedule/space_generator/schedule_fn.py | 2 +-
.../space_generator/space_generator.py | 6 +-
python/tvm/meta_schedule/testing/__init__.py | 8 -
python/tvm/meta_schedule/testing/dummy_object.py | 60 ++++
python/tvm/meta_schedule/testing/utils.py | 109 +------
python/tvm/meta_schedule/tune.py | 329 ++------------------
python/tvm/meta_schedule/tune_context.py | 118 ++++++-
.../search_strategy/evolutionary_search.cc | 13 +-
src/meta_schedule/search_strategy/replay_func.cc | 3 +-
src/meta_schedule/search_strategy/replay_trace.cc | 3 +-
.../search_strategy/search_strategy.cc | 5 +-
src/meta_schedule/task_scheduler/task_scheduler.cc | 107 +------
src/meta_schedule/tune_context.cc | 81 +++++
.../test_meta_schedule_custom_rule_winograd_cpu.py | 20 +-
...test_meta_schedule_custom_rule_winograd_cuda.py | 17 +-
.../unittest/test_meta_schedule_integration.py | 38 ++-
.../test_meta_schedule_measure_callback.py | 75 +++--
.../unittest/test_meta_schedule_multi_anchor.py | 27 +-
...eta_schedule_mutator_mutate_compute_location.py | 12 +-
.../test_meta_schedule_mutator_mutate_parallel.py | 12 +-
..._meta_schedule_mutator_mutate_thread_binding.py | 12 +-
.../test_meta_schedule_mutator_mutate_tile_size.py | 10 +-
.../test_meta_schedule_mutator_mutate_unroll.py | 12 +-
.../test_meta_schedule_post_order_apply.py | 43 +--
...meta_schedule_postproc_disallow_dynamic_loop.py | 3 +-
..._schedule_postproc_rewrite_cooperative_fetch.py | 3 +-
...ta_schedule_postproc_rewrite_reduction_block.py | 3 +-
...est_meta_schedule_postproc_rewrite_tensorize.py | 6 +-
...meta_schedule_postproc_rewrite_unbound_block.py | 3 +-
.../test_meta_schedule_postproc_verify_gpu_code.py | 3 +-
...test_meta_schedule_schedule_rule_add_rfactor.py | 4 +-
.../test_meta_schedule_schedule_rule_auto_bind.py | 4 +-
...test_meta_schedule_schedule_rule_auto_inline.py | 4 +-
...chedule_schedule_rule_cross_thread_reduction.py | 9 +-
...ta_schedule_schedule_rule_multi_level_tiling.py | 15 +-
...dule_schedule_rule_parallel_vectorize_unroll.py | 4 +-
...hedule_schedule_rule_random_compute_location.py | 6 +-
.../unittest/test_meta_schedule_search_strategy.py | 107 ++++---
.../unittest/test_meta_schedule_space_generator.py | 16 +-
.../unittest/test_meta_schedule_task_scheduler.py | 142 +++++----
.../unittest/test_meta_schedule_tune_relay.py | 51 ++-
52 files changed, 1111 insertions(+), 886 deletions(-)
create mode 100644 python/tvm/meta_schedule/database/memory_database.py
create mode 100644 python/tvm/meta_schedule/default_config.py
create mode 100644 python/tvm/meta_schedule/testing/dummy_object.py
