You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2021/05/28 17:44:31 UTC
[tvm] branch main updated (f0aedc4 -> ece644c)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from f0aedc4 Pin black version (#8139)
add ece644c [IR][Pass][Instrument] Pass instrument framework (#7952)
No new revisions were added by this update.
Summary of changes:
include/tvm/ir/instrument.h | 157 ++++++
include/tvm/ir/transform.h | 54 ++-
python/tvm/__init__.py | 2 +-
python/tvm/ir/__init__.py | 1 +
.../_ffi_api.py => ir/_ffi_instrument_api.py} | 5 +-
python/tvm/ir/instrument.py | 159 +++++++
python/tvm/ir/transform.py | 51 +-
src/ir/instrument.cc | 341 +++++++++++++
src/ir/transform.cc | 245 ++++------
src/relay/ir/transform.cc | 4 -
src/tir/ir/transform.cc | 3 -
tests/python/relay/test_pass_annotation.py | 15 +-
tests/python/relay/test_pass_instrument.py | 530 +++++++++++++++++++++
tests/python/relay/test_pass_manager.py | 37 +-
tests/python/relay/test_pass_profiler.py | 41 --
tutorials/dev/use_pass_infra.py | 8 +-
16 files changed, 1384 insertions(+), 269 deletions(-)
create mode 100644 include/tvm/ir/instrument.h
copy python/tvm/{arith/_ffi_api.py => ir/_ffi_instrument_api.py} (91%)
create mode 100644 python/tvm/ir/instrument.py
create mode 100644 src/ir/instrument.cc
create mode 100644 tests/python/relay/test_pass_instrument.py
delete mode 100644 tests/python/relay/test_pass_profiler.py