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 14751b3491 [relax][tflite] Add PRELU/LRN/SQUARED_DIFFERENCE tests
(partial #18971) (#19404)
add 1b94055eb6 Fix #18714: [Docs] python -c "import tvm; print(tvm.file)"
fail (#19407)
add b3439430e7 [Relax][Frontend][KVCache] Restructure kv_cache kernels
(#19405)
add 9d13fc04d2 [S-TIR] Fix Segfault when applying Parallel during TIR
schedule rewriting (#19403)
add 0a79095b1d [S-TIR] Fix cache_read/cache_write region when inner block
has T.whe… (#19406)
No new revisions were added by this update.
Summary of changes:
docs/install/from_source.rst | 3 +-
.../tvm/relax/frontend/nn/llm/_decode_kernels.py | 526 ++++
python/tvm/relax/frontend/nn/llm/_kernel_common.py | 520 ++++
python/tvm/relax/frontend/nn/llm/_page_kernels.py | 293 ++
.../tvm/relax/frontend/nn/llm/_prefill_kernels.py | 1017 +++++++
python/tvm/relax/frontend/nn/llm/kv_cache.py | 2941 +-------------------
python/tvm/relax/frontend/nn/llm/tree_attn.py | 125 +-
.../postproc/rewrite_parallel_vectorize_unroll.cc | 23 +-
src/s_tir/schedule/primitive/cache_read_write.cc | 111 +-
...e_postproc_rewrite_parallel_vectorize_unroll.py | 24 +
.../schedule/test_tir_schedule_cache_read_write.py | 170 ++
11 files changed, 2782 insertions(+), 2971 deletions(-)
create mode 100644 python/tvm/relax/frontend/nn/llm/_decode_kernels.py
create mode 100644 python/tvm/relax/frontend/nn/llm/_kernel_common.py
create mode 100644 python/tvm/relax/frontend/nn/llm/_page_kernels.py
create mode 100644 python/tvm/relax/frontend/nn/llm/_prefill_kernels.py
