You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ju...@apache.org on 2021/09/25 17:05:09 UTC
[tvm] branch main updated (b33a1a7 -> 80de123)
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 b33a1a7 [TensorIR][Bugfix] Disallow fusing loops with dependency (#9112)
add 80de123 [Meta Schedule][M3a] SpaceGenerator (#9079)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/space_generator.h | 122 +++++++++++++++++++++
{src => include/tvm}/meta_schedule/tune_context.h | 6 +
python/tvm/meta_schedule/__init__.py | 1 +
.../space_generator/__init__.py} | 12 +-
.../meta_schedule/space_generator/schedule_fn.py | 90 +++++++++++++++
.../space_generator/space_generator.py | 93 ++++++++++++++++
.../space_generator/space_generator_union.py} | 32 ++++--
python/tvm/meta_schedule/tune_context.py | 9 +-
.../space_generator/space_generator.cc | 44 ++++++++
.../space_generator/space_generator_union.cc | 70 ++++++++++++
src/meta_schedule/tune_context.cc | 10 +-
src/meta_schedule/utils.h | 4 +-
.../unittest/test_meta_schedule_space_generator.py | 88 +++++++++++++++
.../unittest/test_meta_schedule_tune_context.py | 2 +-
14 files changed, 559 insertions(+), 24 deletions(-)
create mode 100644 include/tvm/meta_schedule/space_generator.h
rename {src => include/tvm}/meta_schedule/tune_context.h (89%)
copy python/tvm/{relay/_make.py => meta_schedule/space_generator/__init__.py} (72%)
create mode 100644 python/tvm/meta_schedule/space_generator/schedule_fn.py
create mode 100644 python/tvm/meta_schedule/space_generator/space_generator.py
copy python/tvm/{relay/transform/infer_layout_utils.py => meta_schedule/space_generator/space_generator_union.py} (54%)
mode change 100755 => 100644
create mode 100644 src/meta_schedule/space_generator/space_generator.cc
create mode 100644 src/meta_schedule/space_generator/space_generator_union.cc
create mode 100644 tests/python/unittest/test_meta_schedule_space_generator.py