This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 9849b89 [TIR] Change the behavior of read/write region analysis for
reduction blocks. (#10638)
add 5679bd2 [TIR][Schedule] Transform layout (#10538)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/index_map.h | 16 ++
include/tvm/tir/schedule/schedule.h | 24 ++
python/tvm/tir/__init__.py | 2 +-
python/tvm/tir/function.py | 58 ++++-
python/tvm/tir/schedule/__init__.py | 2 +-
python/tvm/tir/schedule/schedule.py | 87 +++++++-
src/tir/ir/index_map.cc | 51 ++++-
src/tir/schedule/analysis.h | 10 +
src/tir/schedule/analysis/analysis.cc | 31 +++
src/tir/schedule/concrete_schedule.cc | 10 +
src/tir/schedule/concrete_schedule.h | 4 +-
src/tir/schedule/primitive.h | 16 ++
src/tir/schedule/primitive/block_annotate.cc | 38 ----
.../schedule/primitive/layout_transformation.cc | 241 +++++++++++++++++++++
src/tir/schedule/schedule.cc | 8 +
src/tir/schedule/traced_schedule.cc | 14 ++
src/tir/schedule/traced_schedule.h | 3 +
.../unittest/test_tir_schedule_transform_layout.py | 125 +++++++++++
18 files changed, 696 insertions(+), 44 deletions(-)
create mode 100644 src/tir/schedule/primitive/layout_transformation.cc
create mode 100644 tests/python/unittest/test_tir_schedule_transform_layout.py
