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 133bb9c [microNPU][2c] Add performance modelling to cascader (#9778)
add 77c66f0 [MetaSchedule] random compute location (#9940)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/schedule/schedule.h | 8 ++
include/tvm/tir/stmt.h | 7 +
python/tvm/meta_schedule/schedule_rule/__init__.py | 1 +
.../random_compute_location.py} | 13 +-
.../tvm/meta_schedule/testing/space_generation.py | 29 ++--
python/tvm/tir/schedule/schedule.py | 26 ++++
.../schedule_rule/random_compute_location.cc | 123 +++++++++++++++++
src/tir/schedule/analysis.h | 33 +++++
src/tir/schedule/analysis/analysis.cc | 152 +++++++++++++++++++++
src/tir/schedule/concrete_schedule.cc | 9 ++
src/tir/schedule/concrete_schedule.h | 2 +
src/tir/schedule/primitive.h | 11 ++
src/tir/schedule/primitive/sampling.cc | 64 +++++++++
src/tir/schedule/schedule.cc | 2 +
src/tir/schedule/traced_schedule.cc | 14 ++
src/tir/schedule/traced_schedule.h | 1 +
src/tir/schedule/utils.h | 12 ++
...hedule_schedule_rule_random_compute_location.py | 93 +++++++++++++
.../python/unittest/test_tir_schedule_sampling.py | 77 +++++++++++
19 files changed, 656 insertions(+), 21 deletions(-)
copy python/tvm/meta_schedule/{measure_callback/add_to_database.py =>
schedule_rule/random_compute_location.py} (70%)
copy apps/dso_plugin_module/Makefile =>
python/tvm/meta_schedule/testing/space_generation.py (53%)
create mode 100644 src/meta_schedule/schedule_rule/random_compute_location.cc
create mode 100644
tests/python/unittest/test_meta_schedule_schedule_rule_random_compute_location.py
