This is an automated email from the ASF dual-hosted git repository.
syfeng pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 02172c3a5e [FIX][RELAX][ONNX] Fix typo in onnx frontend (#17467)
add 35d6a1b9d2 [TIR][Schedule] Add annotate_buffer_access primitive
(#17423)
No new revisions were added by this update.
Summary of changes:
include/tvm/tir/schedule/schedule.h | 11 +
include/tvm/tir/stmt.h | 10 +
python/tvm/tir/schedule/schedule.py | 136 +++++++++
src/tir/schedule/concrete_schedule.cc | 10 +
src/tir/schedule/concrete_schedule.h | 2 +
src/tir/schedule/primitive.h | 10 +
.../schedule/primitive/annotate_buffer_access.cc | 167 +++++++++++
src/tir/schedule/schedule.cc | 7 +
src/tir/schedule/traced_schedule.cc | 12 +
src/tir/schedule/traced_schedule.h | 2 +
src/tir/transforms/compact_buffer_region.cc | 43 ++-
.../test_tir_schedule_annotate_buffer_access.py | 332 +++++++++++++++++++++
12 files changed, 736 insertions(+), 6 deletions(-)
create mode 100644 src/tir/schedule/primitive/annotate_buffer_access.cc
create mode 100644
tests/python/tir-schedule/test_tir_schedule_annotate_buffer_access.py
