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 1fcb62023f [WebGPU] Fix unexpected device lost error when intentional
dispose (#17250)
add 77391714ab Replacing unary ops with LookUpTable and Take op to improve
performance (#17214)
add b40a02c265 [Relax] Add KVCache Interface for Relax NNModule (#17261)
add 6ae29610a5 [ROCm] Support ROCm 6 (#17256)
No new revisions were added by this update.
Summary of changes:
cmake/modules/ROCM.cmake | 1 +
cmake/utils/FindRCCL.cmake | 2 +-
python/tvm/contrib/hexagon/generate_take_op.py | 98 +
python/tvm/contrib/hexagon/hexagon_unary_ops.py | 97 +
.../tvm/relax/frontend/nn/llm}/__init__.py | 10 +-
.../tvm/relax/frontend/nn/llm/kv_cache.py | 2085 +++++---------------
.../relax/frontend/nn/llm/position_embedding.py | 287 +++
python/tvm/relax/frontend/nn/llm/tree_attn.py | 411 ++++
src/runtime/rocm/rocm_device_api.cc | 2 +-
tests/python/contrib/test_hexagon/test_take.py | 393 ++++
...runtime_builtin_paged_attention_kv_cache_tir.py | 1765 +----------------
11 files changed, 1845 insertions(+), 3306 deletions(-)
create mode 100644 python/tvm/contrib/hexagon/generate_take_op.py
create mode 100644 python/tvm/contrib/hexagon/hexagon_unary_ops.py
copy {apps/uma/_template => python/tvm/relax/frontend/nn/llm}/__init__.py (78%)
copy tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_tir.py
=> python/tvm/relax/frontend/nn/llm/kv_cache.py (50%)
create mode 100644 python/tvm/relax/frontend/nn/llm/position_embedding.py
create mode 100644 python/tvm/relax/frontend/nn/llm/tree_attn.py
create mode 100644 tests/python/contrib/test_hexagon/test_take.py
