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 5012462 [TensorIR][M2a] Reduction Factoring (RFactor) (#8544)
add 2a8950b [TensorIR] Support for match_buffer from subregion (#8585)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/transform.h | 6 +
python/tvm/script/context_maintainer.py | 6 +-
python/tvm/script/parser.py | 22 +-
python/tvm/script/special_stmt.py | 98 ++---
python/tvm/tir/buffer.py | 2 +-
python/tvm/tir/transform/transform.py | 11 +
src/driver/driver_api.cc | 1 +
src/printer/tir_text_printer.cc | 4 +-
src/printer/tvmscript_printer.cc | 25 +-
src/tir/analysis/block_access_region_detector.cc | 90 ++--
src/tir/analysis/buffer_access_lca_detector.cc | 17 +-
src/tir/ir/script/script_complete.cc | 24 +-
src/tir/ir/stmt.cc | 49 ++-
src/tir/transforms/compact_buffer_region.cc | 13 +
src/tir/transforms/ir_utils.cc | 47 +++
src/tir/transforms/ir_utils.h | 16 +
src/tir/transforms/lower_match_buffer.cc | 270 ++++++++++++
tests/python/integration/test_lower.py | 327 +++++++++++++++
.../test_tir_analysis_detect_buffer_access_lca.py | 33 ++
.../test_tir_analysis_get_block_access_region.py | 78 ++++
.../python/unittest/test_tir_lower_match_buffer.py | 455 +++++++++++++++++++++
tests/python/unittest/test_tir_nodes.py | 7 +-
.../unittest/test_tir_schedule_compute_inline.py | 2 +-
.../test_tir_transform_compact_buffer_region.py | 51 +++
.../test_tir_transform_lower_init_block.py | 37 ++
...sform_plan_update_buffer_allocation_location.py | 27 ++
tests/python/unittest/test_tvmscript_complete.py | 51 ++-
.../python/unittest/test_tvmscript_error_report.py | 20 +-
tests/python/unittest/test_tvmscript_roundtrip.py | 42 +-
29 files changed, 1670 insertions(+), 161 deletions(-)
create mode 100644 src/tir/transforms/lower_match_buffer.cc
create mode 100644 tests/python/integration/test_lower.py
create mode 100644 tests/python/unittest/test_tir_lower_match_buffer.py
