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 6c37194c2f [Relax][PyTorch] Support lstm op for ExportedProgram
importer (#18346)
add c00c66259a [Relax][ONNX] Support AllClassNMS Operator for ONNX
Frontend (#18321)
add b129c95742 [FFI][ABI] Bump tvm-ffi to latest (#18354)
add 3015acd7e6 [CUDA] Update FlashInfer JIT integration (#18353)
No new revisions were added by this update.
Summary of changes:
3rdparty/tvm-ffi | 2 +-
include/tvm/relax/attrs/vision.h | 54 +++
python/tvm/relax/backend/cuda/flashinfer.py | 481 +++++---------------
python/tvm/relax/frontend/nn/llm/kv_cache.py | 21 +-
python/tvm/relax/frontend/onnx/onnx_frontend.py | 179 +++++++-
python/tvm/relax/op/__init__.py | 1 +
python/tvm/relax/op/op_attrs.py | 5 +
.../msc/core => relax/op/vision}/__init__.py | 3 +-
python/tvm/relax/op/{image => vision}/_ffi_api.py | 2 +-
python/tvm/relax/op/vision/nms.py | 75 ++++
.../tvm/relax/transform/legalize_ops/__init__.py | 1 +
python/tvm/relax/transform/legalize_ops/vision.py | 120 +++++
python/tvm/script/ir_builder/relax/ir.py | 2 +
python/tvm/topi/__init__.py | 1 +
python/tvm/topi/cpp/vision/__init__.py | 1 +
.../{contrib/msc/core => topi/vision}/__init__.py | 3 +-
python/tvm/topi/vision/nms.py | 500 +++++++++++++++++++++
python/tvm/topi/vision/nms_util.py | 473 +++++++++++++++++++
src/relax/ir/emit_te.h | 4 +
src/relax/op/vision/nms.cc | 114 +++++
src/relax/op/{nn/attention.h => vision/nms.h} | 25 +-
src/runtime/vm/attn_backend.cc | 11 +-
src/runtime/vm/attn_backend.h | 213 +++++++--
src/runtime/vm/attn_utils.h | 34 +-
src/runtime/vm/paged_kv_cache.cc | 2 +-
src/te/operation/create_primfunc.cc | 5 +-
tests/python/relax/test_frontend_onnx.py | 426 ++++++++++++++++++
tests/python/relax/test_group_gemm_flashinfer.py | 39 +-
tests/python/relax/test_op_vision.py | 90 ++++
..._builtin_paged_attention_kv_cache_flashinfer.py | 71 ++-
...ltin_paged_attention_kv_cache_mla_flashinfer.py | 69 ++-
...p_set.py => test_tvmscript_parser_op_vision.py} | 42 +-
32 files changed, 2491 insertions(+), 578 deletions(-)
create mode 100644 include/tvm/relax/attrs/vision.h
copy python/tvm/{contrib/msc/core => relax/op/vision}/__init__.py (94%)
copy python/tvm/relax/op/{image => vision}/_ffi_api.py (94%)
create mode 100644 python/tvm/relax/op/vision/nms.py
create mode 100644 python/tvm/relax/transform/legalize_ops/vision.py
copy python/tvm/{contrib/msc/core => topi/vision}/__init__.py (94%)
create mode 100644 python/tvm/topi/vision/nms.py
create mode 100644 python/tvm/topi/vision/nms_util.py
create mode 100644 src/relax/op/vision/nms.cc
copy src/relax/op/{nn/attention.h => vision/nms.h} (58%)
create mode 100644 tests/python/relax/test_op_vision.py
copy tests/python/relax/{test_tvmscript_parser_op_set.py =>
test_tvmscript_parser_op_vision.py} (54%)
