You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by gi...@apache.org on 2024/01/25 05:02:57 UTC
(tvm) branch nightly updated (7eb45ee5db -> 593a4bd9d7)
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 7eb45ee5db [Schedule] Loop-Partition Scheduling Primitive (#16431)
add a86e41bcd2 [Unity][TVMScript] Update `call_packed` semantics to support empty sinfo_args (#16379)
add 1ee0cff6f9 [Minor] Fix Docstring for sphinx-build (#16461)
add 20b08a5c8d [Runtime][KVCache] AttentionWithFusedQKV and RoPE mode (#16456)
add 0e8e421766 [Cherry-pick][MSC][M4.1] Add plugin && plugin_builder, enable build and test in different frameworks (#16397) (#16460)
add 593a4bd9d7 [Relax] NDArray Cache Update with DLTensor Support (#16464)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/msc/core/codegen/codegen.py | 33 +-
.../contrib/msc/framework/tvm/codegen/codegen.py | 30 +-
.../tvm/contrib/msc/plugin}/__init__.py | 3 +-
.../tvm/{arith => contrib/msc/plugin}/_ffi_api.py | 6 +-
python/tvm/contrib/msc/plugin/build.py | 286 +++++
.../{framework/tvm => plugin}/codegen/__init__.py | 2 +-
python/tvm/contrib/msc/plugin/codegen/codegen.py | 319 ++++++
python/tvm/contrib/msc/plugin/codegen/sources.py | 1157 ++++++++++++++++++++
.../{debugger => msc/plugin/op}/__init__.py | 1 +
.../{arith => contrib/msc/plugin/op}/_ffi_api.py | 6 +-
python/tvm/contrib/msc/plugin/register.py | 85 ++
python/tvm/contrib/msc/plugin/utils.py | 109 ++
python/tvm/relax/op/nn/nn.py | 43 +-
python/tvm/script/ir_builder/relax/ir.py | 6 +-
src/contrib/msc/core/ir/plugin.cc | 327 ++++++
src/contrib/msc/core/ir/plugin.h | 686 ++++++++++++
src/contrib/msc/core/utils.cc | 41 +
src/contrib/msc/core/utils.h | 28 +
src/contrib/msc/plugin/base_codegen.h | 674 ++++++++++++
src/contrib/msc/plugin/codegen_utils.h | 75 ++
src/contrib/msc/plugin/tensorrt_codegen.cc | 901 +++++++++++++++
src/contrib/msc/plugin/tensorrt_codegen.h | 134 +++
src/contrib/msc/plugin/torch_codegen.cc | 510 +++++++++
src/contrib/msc/plugin/torch_codegen.h | 137 +++
src/contrib/msc/plugin/tvm_codegen.cc | 411 +++++++
src/contrib/msc/plugin/tvm_codegen.h | 124 +++
src/runtime/relax_vm/kv_cache.h | 13 +
src/runtime/relax_vm/ndarray_cache_support.cc | 23 +-
src/runtime/relax_vm/paged_kv_cache.cc | 308 ++++--
tests/python/contrib/test_msc/test_plugin.py | 309 ++++++
..._builtin_paged_attention_kv_cache_flashinfer.py | 230 +++-
...runtime_builtin_paged_attention_kv_cache_tir.py | 296 ++++-
tests/python/relax/test_tvmscript_parser.py | 22 +
33 files changed, 7054 insertions(+), 281 deletions(-)
copy {tests/python/contrib/test_hexagon => python/tvm/contrib/msc/plugin}/__init__.py (94%)
copy python/tvm/{arith => contrib/msc/plugin}/_ffi_api.py (90%)
create mode 100644 python/tvm/contrib/msc/plugin/build.py
copy python/tvm/contrib/msc/{framework/tvm => plugin}/codegen/__init__.py (94%)
create mode 100644 python/tvm/contrib/msc/plugin/codegen/codegen.py
create mode 100644 python/tvm/contrib/msc/plugin/codegen/sources.py
copy python/tvm/contrib/{debugger => msc/plugin/op}/__init__.py (96%)
copy python/tvm/{arith => contrib/msc/plugin/op}/_ffi_api.py (90%)
create mode 100644 python/tvm/contrib/msc/plugin/register.py
create mode 100644 python/tvm/contrib/msc/plugin/utils.py
create mode 100644 src/contrib/msc/core/ir/plugin.cc
create mode 100644 src/contrib/msc/core/ir/plugin.h
create mode 100644 src/contrib/msc/plugin/base_codegen.h
create mode 100644 src/contrib/msc/plugin/codegen_utils.h
create mode 100644 src/contrib/msc/plugin/tensorrt_codegen.cc
create mode 100644 src/contrib/msc/plugin/tensorrt_codegen.h
create mode 100644 src/contrib/msc/plugin/torch_codegen.cc
create mode 100644 src/contrib/msc/plugin/torch_codegen.h
create mode 100644 src/contrib/msc/plugin/tvm_codegen.cc
create mode 100644 src/contrib/msc/plugin/tvm_codegen.h
create mode 100644 tests/python/contrib/test_msc/test_plugin.py