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 c8a892b [Vulkan] Rewrote PointerValueTypeRewrite transform (#8528)
add 5012462 [TensorIR][M2a] Reduction Factoring (RFactor) (#8544)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/analysis.h | 22 +-
include/tvm/tir/schedule/block_scope.h | 2 +-
include/tvm/tir/schedule/schedule.h | 18 +
include/tvm/tir/stmt.h | 19 +
python/tvm/script/special_stmt.py | 2 +-
python/tvm/tir/schedule/schedule.py | 147 +++-
src/arith/canonical_simplify.cc | 6 +-
src/arith/detect_linear_equation.cc | 4 +-
src/printer/tir_text_printer.cc | 17 -
src/printer/tvmscript_printer.cc | 17 -
src/te/autodiff/ad_simplify.cc | 4 +-
src/te/operation/compute_op.cc | 2 +-
src/te/operation/tensorize.cc | 4 +-
src/tir/analysis/var_touch.cc | 30 +-
src/tir/ir/expr.cc | 3 -
src/tir/schedule/analysis.h | 123 ++-
src/tir/schedule/analysis/analysis.cc | 643 +++++++++++++-
src/tir/schedule/concrete_schedule.cc | 10 +
src/tir/schedule/concrete_schedule.h | 7 +-
src/tir/schedule/primitive.h | 11 +
src/tir/schedule/primitive/compute_inline.cc | 6 +-
src/tir/schedule/primitive/reduction.cc | 963 +++++++++++++++++++++
src/tir/schedule/schedule.cc | 5 +
src/tir/schedule/state.cc | 32 +-
src/tir/schedule/utils.h | 27 +
src/tir/transforms/loop_partition.cc | 19 +-
src/tir/transforms/lower_warp_memory.cc | 2 +-
tests/python/unittest/test_te_schedule.py | 2 +-
.../python/unittest/test_tir_schedule_reduction.py | 674 ++++++++++++++
29 files changed, 2656 insertions(+), 165 deletions(-)
create mode 100644 src/tir/schedule/primitive/reduction.cc
create mode 100644 tests/python/unittest/test_tir_schedule_reduction.py
