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 11f2253b9c Restore "pytest.mark.gpu" for RELAX tests (#16741)
add 2f395f1756 [SVE][TOPI] Add conv2d NHWC hybrid SVE schedule for
`arm_cpu` (#16899)
add 5cf4ca6d15 [Marvell BYOC]: Marvell AI Accelerator Integration - Phase
2 (#16915)
add 4f8c03fad3 [TVMScript] Support `T.launch_thread` with i64 dtype
(#16916)
No new revisions were added by this update.
Summary of changes:
cmake/modules/contrib/Mrvl.cmake | 1 +
docker/Dockerfile.demo_mrvl | 28 +++
docs/how_to/deploy/mrvl.rst | 89 ++++----
include/tvm/script/ir_builder/tir/ir.h | 3 +-
python/tvm/contrib/mrvl.py | 172 +++++++++++++++
python/tvm/relay/op/contrib/mrvl.py | 80 ++++---
python/tvm/relay/op/strategy/arm_cpu.py | 12 ++
python/tvm/script/ir_builder/tir/ir.py | 7 +-
python/tvm/topi/arm_cpu/arm_utils.py | 99 ++++++++-
python/tvm/topi/arm_cpu/conv2d.py | 43 +++-
python/tvm/topi/arm_cpu/conv2d_gemm.py | 134 ++++++------
python/tvm/topi/nn/conv2d.py | 7 +-
src/arith/analyzer.cc | 3 +-
src/arith/const_int_bound.cc | 3 +
src/arith/int_set.cc | 6 +
src/arith/rewrite_simplify.cc | 20 ++
src/arith/scalable_expression.cc | 5 +
src/arith/scalable_expression.h | 7 +
src/relay/backend/contrib/mrvl/codegen.cc | 231 ++++++++++++---------
src/relay/backend/contrib/mrvl/compiler_attr.cc | 1 -
src/relay/backend/utils.cc | 4 +-
src/runtime/contrib/mrvl/mrvl_base64.h | 78 +++++++
src/runtime/contrib/mrvl/mrvl_runtime.cc | 38 +++-
src/runtime/contrib/mrvl/mrvl_sw_runtime_lib.cc | 175 ++++++++++++++++
.../runtime/contrib/mrvl/mrvl_sw_runtime_lib.h | 21 +-
src/script/ir_builder/tir/ir.cc | 10 +-
src/tir/transforms/storage_rewrite.cc | 7 +
tests/python/arith/test_arith_rewrite_simplify.py | 25 +++
.../python/codegen/test_target_codegen_aarch64.py | 35 ++++
tests/python/contrib/test_mrvl/infrastructure.py | 50 ++++-
tests/python/contrib/test_mrvl/test_mrvl.py | 49 +++--
.../tir-schedule/test_tir_schedule_split_fuse.py | 86 ++++----
.../test_tir_transform_inject_ptx_async_copy.py | 4 +-
tests/python/topi/test_topi_conv2d_nhwc.py | 47 +++--
.../python/tvmscript/test_tvmscript_parser_tir.py | 15 ++
35 files changed, 1237 insertions(+), 358 deletions(-)
create mode 100644 src/runtime/contrib/mrvl/mrvl_base64.h
create mode 100644 src/runtime/contrib/mrvl/mrvl_sw_runtime_lib.cc
copy include/tvm/runtime/contrib/libtorch_runtime.h =>
src/runtime/contrib/mrvl/mrvl_sw_runtime_lib.h (64%)
