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 541f9c280c [Rocm] Fix non-standard rocm path (#17295)
add 47e964a597 [Codegen][WebGPU] LetNode common subexpr override (#17302)
add 384360f628 [Relax][Bugfix] Support torch.unbind op and fix bugs for
expand && split (#17292)
add d5d5ebb601 [Support] Fix the Read/Write of socket stream (#17284)
add c4acc79bde [Relax] Avoid wrapping TupleStructInfo into a Tuple for
R.call_tir (#17243)
add c61982e2cd [TE][CreatePrimFunc] Fix create reduce block with spatial
iter dependent init value (#17301)
add 3138328207 [Runtime] Support KV cache with RoPE extension factor array
(#17294)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/msc/core/frontend/translate.py | 2 +
python/tvm/relax/frontend/torch/fx_translator.py | 33 +++++-
python/tvm/relax/op/base.py | 37 ++++--
src/runtime/relax_vm/kv_state.h | 1 +
src/runtime/relax_vm/paged_kv_cache.cc | 63 ++++++----
src/support/socket.h | 4 +-
src/target/source/codegen_webgpu.cc | 21 ++++
src/target/source/codegen_webgpu.h | 3 +-
src/te/operation/create_primfunc.cc | 17 ++-
tests/python/contrib/test_msc/test_graph_build.py | 54 ++++++++-
tests/python/contrib/test_msc/test_pipeline.py | 2 +-
.../contrib/test_msc/test_translate_relax.py | 41 ++++++-
.../contrib/test_msc/test_translate_relay.py | 34 +++++-
.../contrib/test_msc/test_translate_tensorrt.py | 36 +++++-
.../contrib/test_msc/test_translate_torch.py | 35 +++++-
tests/python/relax/test_frontend_from_fx.py | 128 ++++++++++++++++++++-
..._builtin_paged_attention_kv_cache_flashinfer.py | 3 +
...runtime_builtin_paged_attention_kv_cache_tir.py | 1 +
tests/python/relax/test_tvmscript_parser.py | 36 ++++++
tests/python/te/test_te_create_primfunc.py | 73 ++++++++++++
20 files changed, 552 insertions(+), 72 deletions(-)