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/11/05 04:45:32 UTC
[tvm] branch main updated (b8fb438 -> 85fd85f)
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 b8fb438 Fix several typos in pytest_target_parameterization.rst (#9447)
add 85fd85f [MetaSchedule] Task Extraction (#9382)
No new revisions were added by this update.
Summary of changes:
include/tvm/meta_schedule/integration.h | 214 ++++++++++++++++++
python/tvm/meta_schedule/__init__.py | 1 +
python/tvm/meta_schedule/integration.py | 250 +++++++++++++++++++++
.../_ctypes => meta_schedule/testing}/__init__.py | 4 +-
.../{testing.py => testing/local_rpc.py} | 2 +-
python/tvm/meta_schedule/testing/relay_workload.py | 88 ++++++++
python/tvm/te/__init__.py | 2 +-
python/tvm/te/operation.py | 26 ++-
src/meta_schedule/integration.cc | 151 +++++++++++++
src/relay/backend/te_compiler.cc | 25 ++-
src/relay/backend/te_compiler_cache.cc | 49 ++--
src/relay/backend/te_compiler_cache.h | 11 +-
src/relay/backend/utils.h | 9 +
src/te/operation/create_primfunc.cc | 38 +++-
.../unittest/test_meta_schedule_integration.py | 120 ++++++++++
15 files changed, 954 insertions(+), 36 deletions(-)
create mode 100644 include/tvm/meta_schedule/integration.h
create mode 100644 python/tvm/meta_schedule/integration.py
copy python/tvm/{_ffi/_ctypes => meta_schedule/testing}/__init__.py (87%)
rename python/tvm/meta_schedule/{testing.py => testing/local_rpc.py} (97%)
create mode 100644 python/tvm/meta_schedule/testing/relay_workload.py
create mode 100644 src/meta_schedule/integration.cc
create mode 100644 tests/python/unittest/test_meta_schedule_integration.py