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 5814fdd prune dnnl subgraph, and add related test case. (#10835)
add 5629f8a [MetaSchedule] Add Gradient Based Task Scheduler (#10366)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/search_strategy.h | 12 +-
include/tvm/meta_schedule/task_scheduler.h | 94 +++++----
include/tvm/meta_schedule/tune_context.h | 4 +-
include/tvm/support/random_engine.h | 18 +-
include/tvm/tir/schedule/schedule.h | 2 +-
.../search_strategy/evolutionary_search.py | 13 +-
.../meta_schedule/search_strategy/replay_func.py | 13 +-
.../meta_schedule/search_strategy/replay_trace.py | 16 +-
.../tvm/meta_schedule/task_scheduler/__init__.py | 1 +
.../{round_robin.py => gradient_based.py} | 61 +++---
.../meta_schedule/task_scheduler/round_robin.py | 15 +-
.../meta_schedule/task_scheduler/task_scheduler.py | 86 ++++----
.../testing/tune_relay_meta_schedule.py | 8 +-
.../meta_schedule/testing/tune_te_meta_schedule.py | 3 +-
python/tvm/meta_schedule/tune.py | 14 +-
python/tvm/meta_schedule/utils.py | 2 +-
.../measure_callback/echo_statistics.cc | 10 +-
.../search_strategy/evolutionary_search.cc | 16 +-
src/meta_schedule/search_strategy/replay_func.cc | 12 +-
src/meta_schedule/search_strategy/replay_trace.cc | 12 +-
src/meta_schedule/task_scheduler/gradient_based.cc | 228 +++++++++++++++++++++
src/meta_schedule/task_scheduler/round_robin.cc | 10 +-
src/meta_schedule/task_scheduler/task_scheduler.cc | 134 ++++++------
src/meta_schedule/tune_context.cc | 5 +-
src/meta_schedule/utils.h | 23 +++
src/support/table_printer.h | 154 ++++++++++++++
src/tir/schedule/concrete_schedule.cc | 3 -
src/tir/schedule/concrete_schedule.h | 2 +-
.../test_meta_schedule_measure_callback.py | 9 +-
.../unittest/test_meta_schedule_search_strategy.py | 16 +-
.../unittest/test_meta_schedule_task_scheduler.py | 138 +++++++++----
.../unittest/test_meta_schedule_tune_relay.py | 23 ++-
.../python/unittest/test_meta_schedule_tune_te.py | 3 +-
.../python/unittest/test_meta_schedule_tune_tir.py | 25 ++-
34 files changed, 835 insertions(+), 350 deletions(-)
copy python/tvm/meta_schedule/task_scheduler/{round_robin.py =>
gradient_based.py} (62%)
create mode 100644 src/meta_schedule/task_scheduler/gradient_based.cc
create mode 100644 src/support/table_printer.h
