This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from e3faa55573 [JVM] Align Java GraphModule Initialization with Python API
(#17464)
add 0c67cd8d29 Revert "[KVCACHE] Improved schedule for prefill attention"
(#17466)
add 02172c3a5e [FIX][RELAX][ONNX] Fix typo in onnx frontend (#17467)
add 35d6a1b9d2 [TIR][Schedule] Add annotate_buffer_access primitive
(#17423)
add 58a43c8724 [MetaSchedule] Fix a multilevel tiling error on dynamic
relax workload (#17465)
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/relax/frontend/nn/llm/kv_cache.py | 60 +---
python/tvm/relax/frontend/onnx/onnx_frontend.py | 2 +-
python/tvm/tir/schedule/schedule.py | 136 +++++++++
src/tir/schedule/analysis/analysis.cc | 4 +-
src/tir/schedule/concrete_schedule.cc | 14 +-
src/tir/schedule/concrete_schedule.h | 14 +-
src/tir/schedule/primitive.h | 10 +
.../schedule/primitive/annotate_buffer_access.cc | 167 +++++++++++
src/tir/schedule/schedule.cc | 7 +
src/tir/schedule/trace.cc | 4 +-
src/tir/schedule/traced_schedule.cc | 20 +-
src/tir/schedule/traced_schedule.h | 2 +
src/tir/transforms/compact_buffer_region.cc | 43 ++-
.../test_tir_schedule_annotate_buffer_access.py | 332 +++++++++++++++++++++
.../tir-schedule/test_tir_schedule_sampling.py | 28 ++
.../tir-schedule/test_tir_schedule_split_fuse.py | 35 +++
18 files changed, 834 insertions(+), 65 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
