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/29 05:40:00 UTC
[tvm] branch main updated (e4946f4 -> a16ccf4)
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 e4946f4 [Frontend][PyTorch] support for quantized conv_transpose2d op (#9133)
add a16ccf4 [Meta Schedule][M3a] SearchStrategy (#9132)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/builder.h | 4 +-
include/tvm/meta_schedule/runner.h | 62 ++++++
include/tvm/meta_schedule/search_strategy.h | 246 +++++++++++++++++++++
include/tvm/meta_schedule/space_generator.h | 38 +++-
include/tvm/support/random_engine.h | 10 +
python/tvm/meta_schedule/__init__.py | 2 +
.../_cy2 => meta_schedule/runner}/__init__.py | 3 +-
.../space_generator_union.py => runner/runner.py} | 42 +++-
.../search_strategy}/__init__.py | 6 +-
.../replay_trace.py} | 38 ++--
.../search_strategy/search_strategy.py | 166 ++++++++++++++
.../vision.cc => meta_schedule/runner/runner.cc} | 30 +--
src/meta_schedule/search_strategy/replay_trace.cc | 148 +++++++++++++
.../search_strategy/search_strategy.cc | 68 ++++++
src/meta_schedule/utils.h | 76 +++++++
src/tir/schedule/concrete_schedule.cc | 4 +-
src/tir/schedule/primitive.h | 8 +
src/tir/schedule/primitive/sampling.cc | 12 +
...or.py => test_meta_schedule_search_strategy.py} | 78 ++++---
19 files changed, 955 insertions(+), 86 deletions(-)
create mode 100644 include/tvm/meta_schedule/runner.h
create mode 100644 include/tvm/meta_schedule/search_strategy.h
copy python/tvm/{_ffi/_cy2 => meta_schedule/runner}/__init__.py (92%)
copy python/tvm/meta_schedule/{space_generator/space_generator_union.py => runner/runner.py} (53%)
copy python/tvm/{contrib/tf_op => meta_schedule/search_strategy}/__init__.py (86%)
copy python/tvm/meta_schedule/{space_generator/space_generator_union.py => search_strategy/replay_trace.py} (53%)
create mode 100644 python/tvm/meta_schedule/search_strategy/search_strategy.py
copy src/{topi/vision.cc => meta_schedule/runner/runner.cc} (61%)
create mode 100644 src/meta_schedule/search_strategy/replay_trace.cc
create mode 100644 src/meta_schedule/search_strategy/search_strategy.cc
copy tests/python/unittest/{test_meta_schedule_space_generator.py => test_meta_schedule_search_strategy.py} (50%)