You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2022/02/23 04:07:38 UTC
[tvm] branch last-successful updated (5956125 -> b5f1dab)
This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch last-successful
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 5956125 [ONNX] only broadcast matmul if the shape has changed (#10321)
add b5f1dab [TIR] Tir constants integration into compilation pipeline (#8509)
No new revisions were added by this update.
Summary of changes:
include/tvm/ir/module.h | 42 +++++++
include/tvm/node/structural_hash.h | 9 ++
include/tvm/relay/executor.h | 7 +-
include/tvm/relay/interpreter.h | 4 +-
include/tvm/relay/runtime.h | 2 +
include/tvm/tir/function.h | 38 +-----
include/tvm/tir/stmt.h | 96 ++++++++++++++-
include/tvm/tir/stmt_functor.h | 4 +
include/tvm/tir/transform.h | 10 ++
python/tvm/script/tir/scope_handler.py | 51 ++++++++
python/tvm/te/__init__.py | 2 +-
python/tvm/te/operation.py | 22 ++++
python/tvm/tir/__init__.py | 11 +-
python/tvm/tir/stmt.py | 34 ++++++
python/tvm/tir/transform/transform.py | 11 ++
python/tvm/topi/nn/dense.py | 4 +-
src/ir/module.cc | 7 ++
src/node/structural_hash.cc | 68 +++++------
src/printer/text_printer.h | 2 +
src/printer/tir_text_printer.cc | 13 ++
src/printer/tvmscript_printer.cc | 65 ++++++++++
src/relay/backend/aot_executor_codegen.cc | 45 ++++---
src/relay/backend/build_module.cc | 29 +----
src/relay/backend/executor.cc | 1 +
src/relay/backend/interpreter.cc | 5 +-
src/relay/backend/te_compiler.cc | 10 +-
src/relay/backend/te_compiler_cache.cc | 67 +++++++----
src/relay/backend/te_compiler_cache.h | 4 +-
src/relay/backend/utils.h | 7 +-
src/relay/ir/transform.cc | 2 +-
src/relay/transforms/device_planner.cc | 4 +-
src/relay/transforms/fold_constant.cc | 7 +-
src/relay/transforms/fuse_ops.cc | 41 ++++++-
src/relay/transforms/split_args.cc | 3 +-
src/target/llvm/codegen_amdgpu.cc | 2 +-
src/target/llvm/codegen_llvm.cc | 13 +-
src/target/llvm/codegen_llvm.h | 1 +
src/target/llvm/codegen_nvptx.cc | 2 +-
src/target/source/codegen_c.cc | 34 +++++-
src/target/source/codegen_c.h | 6 +
src/target/source/codegen_c_host.cc | 75 +-----------
src/target/source/codegen_c_host.h | 4 -
src/target/source/codegen_cuda.cc | 2 +-
src/target/source/codegen_opencl.cc | 3 +-
src/target/source/codegen_opencl.h | 2 +-
src/target/source/codegen_params.cc | 16 +--
src/target/spirv/codegen_spirv.cc | 2 +-
src/tir/analysis/verify_gpu_code.cc | 4 +-
src/tir/ir/function.cc | 8 --
src/tir/ir/stmt.cc | 79 +++++++++++-
src/tir/ir/stmt_functor.cc | 19 +++
src/tir/transforms/bind_params.cc | 133 +++++++++++++++++++++
src/tir/transforms/extract_constants.cc | 115 ++++++++++++++++++
src/tir/transforms/lower_tvm_builtin.cc | 2 +-
src/tir/transforms/lower_warp_memory.cc | 2 +-
.../merge_dynamic_shared_memory_allocations.cc | 2 +-
src/tir/transforms/split_host_device.cc | 5 +
src/tir/transforms/storage_flatten.cc | 2 +-
src/tir/transforms/storage_rewrite.cc | 33 ++++-
tests/cpp/ir_functor_test.cc | 5 +-
tests/lint/python_format.sh | 4 +-
tests/python/relay/aot/aot_test_utils.py | 2 +-
tests/python/relay/test_pass_fuse_ops.py | 41 ++++---
tests/python/unittest/test_link_params.py | 37 ++----
.../unittest/test_micro_model_library_format.py | 24 ++--
tests/python/unittest/test_te_schedule_ops.py | 6 +
tests/python/unittest/test_tir_nodes.py | 7 +-
.../test_tir_transform_extract_constants.py | 64 ++++++++++
tests/python/unittest/test_tvmscript_roundtrip.py | 72 +++++++++++
tests/scripts/setup-pytest-env.sh | 4 +-
70 files changed, 1221 insertions(+), 338 deletions(-)
create mode 100644 src/tir/transforms/bind_params.cc
create mode 100644 src/tir/transforms/extract_constants.cc
create mode 100644 tests/python/unittest/test_tir_transform_extract_constants.py