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 2022/11/17 02:58:04 UTC
[tvm] branch main updated (59abd9e106 -> ad5c811411)
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 59abd9e106 [Build] Fix MSVC compile option /bigobj (#13411)
add ad5c811411 [TIR] Unify index data type when creating prim func (#13327)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/data_type_rewriter.h | 155 ++++++++
include/tvm/tir/stmt.h | 1 +
include/tvm/tir/stmt_functor.h | 50 ---
python/tvm/te/operation.py | 8 +-
src/relay/backend/utils.cc | 2 +-
src/te/operation/create_primfunc.cc | 25 +-
src/te/operation/create_primfunc.h | 8 +-
src/tir/ir/data_type_rewriter.cc | 389 ++++++++++++++++++++-
src/tir/ir/stmt_functor.cc | 1 +
src/tir/schedule/primitive/blockize_tensorize.cc | 15 +
src/tir/transforms/lower_match_buffer.cc | 4 +-
src/tir/transforms/narrow_datatype.cc | 139 ++------
tests/cpp/data_type_rewriter_test.cc | 2 +-
.../test_meta_schedule_relay_integration.py | 36 +-
tests/python/unittest/test_te_create_primfunc.py | 25 +-
.../python/unittest/test_tir_schedule_tensorize.py | 20 +-
.../unittest/test_tir_transform_narrow_datatype.py | 33 +-
17 files changed, 675 insertions(+), 238 deletions(-)
create mode 100644 include/tvm/tir/data_type_rewriter.h