This is an automated email from the ASF dual-hosted git repository.
leandron pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 34570f2 [Onnx] Add Adam (#9002)
add 354019d [3/6] Arm(R) Ethos(TM)-U NPU TIR compiler with conv2d support
(#8806)
No new revisions were added by this update.
Summary of changes:
.../tvm/relay/backend/contrib/ethosu/__init__.py | 1 +
.../relay/backend/contrib/ethosu/te/convolution.py | 2 +-
python/tvm/relay/backend/contrib/ethosu/te/dma.py | 4 +-
.../backend/contrib/ethosu/tir}/__init__.py | 2 +-
.../relay/backend/contrib/ethosu/tir/compiler.py | 199 ++++++++
.../backend/contrib/ethosu/tir/convolution.py | 106 ++++
python/tvm/relay/backend/contrib/ethosu/tir/dma.py | 291 +++++++++++
.../tvm/relay/backend/contrib/ethosu/tir/passes.py | 475 ++++++++++++++++++
.../relay/backend/contrib/ethosu/tir/scheduler.py | 277 +++++++++++
.../tvm/relay/backend/contrib/ethosu/tir/spec.py | 263 ++++++++++
.../relay/backend/contrib/ethosu/tir/transform.py | 61 +++
.../tvm/relay/backend/contrib/ethosu/tir/utils.py | 222 +++++++++
.../backend/contrib/ethosu/tir_to_cs_translator.py | 332 +++++++++++++
python/tvm/relay/backend/contrib/ethosu/util.py | 14 +
.../tvm/relay/backend/contrib/ethosu/vela_api.py | 62 +++
src/relay/backend/contrib/ethosu/compiler_attrs.cc | 73 +++
src/relay/backend/contrib/ethosu/to_te_graph.cc | 234 +++++++++
tests/python/contrib/test_ethosu/infra.py | 117 +++++
.../test_ethosu/test_attr_passing.py} | 33 +-
tests/python/contrib/test_ethosu/test_compiler.py | 48 ++
.../contrib/test_ethosu/test_encode_constants.py | 273 ++++++++++
.../contrib/test_ethosu/test_extract_constants.py | 99 ++++
.../python/contrib/test_ethosu/test_lower_to_te.py | 63 +++
.../contrib/test_ethosu/test_replace_conv2d.py | 547 +++++++++++++++++++++
.../contrib/test_ethosu/test_replace_copy.py | 77 +++
tests/python/contrib/test_ethosu/test_scheduler.py | 144 ++++++
tests/python/contrib/test_ethosu/test_vela_api.py | 101 ++++
27 files changed, 4104 insertions(+), 16 deletions(-)
copy python/tvm/{_ffi/_cy2 => relay/backend/contrib/ethosu/tir}/__init__.py
(94%)
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/compiler.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/convolution.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/dma.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/passes.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/scheduler.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/spec.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/transform.py
create mode 100644 python/tvm/relay/backend/contrib/ethosu/tir/utils.py
create mode 100644
python/tvm/relay/backend/contrib/ethosu/tir_to_cs_translator.py
create mode 100644 src/relay/backend/contrib/ethosu/compiler_attrs.cc
create mode 100644 src/relay/backend/contrib/ethosu/to_te_graph.cc
create mode 100644 tests/python/contrib/test_ethosu/infra.py
copy tests/python/{unittest/test_tir_transform_ir_utils.py =>
contrib/test_ethosu/test_attr_passing.py} (54%)
create mode 100644 tests/python/contrib/test_ethosu/test_compiler.py
create mode 100644 tests/python/contrib/test_ethosu/test_encode_constants.py
create mode 100644 tests/python/contrib/test_ethosu/test_extract_constants.py
create mode 100644 tests/python/contrib/test_ethosu/test_lower_to_te.py
create mode 100644 tests/python/contrib/test_ethosu/test_replace_conv2d.py
create mode 100644 tests/python/contrib/test_ethosu/test_replace_copy.py
create mode 100644 tests/python/contrib/test_ethosu/test_scheduler.py
