You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2021/11/03 20:07:24 UTC
[tvm] branch main updated (e0c1256 -> 19b23b9)
This is an automated email from the ASF dual-hosted git repository.
manupa pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from e0c1256 Removed a manual file handler pitfall (#9435)
add 19b23b9 Arm(R) Ethos(TM)-U NPU Pooling operators support (#9384)
No new revisions were added by this update.
Summary of changes:
.../tvm/relay/backend/contrib/ethosu/legalize.py | 136 ++++++++++++--
.../relay/backend/contrib/ethosu/op/__init__.py | 1 +
.../relay/backend/contrib/ethosu/op/convolution.py | 6 +-
.../relay/backend/contrib/ethosu/op/depthwise.py | 8 +-
.../contrib/ethosu/op/{depthwise.py => pooling.py} | 109 +++++-------
.../relay/backend/contrib/ethosu/te/__init__.py | 1 +
.../relay/backend/contrib/ethosu/te/convolution.py | 2 +-
.../relay/backend/contrib/ethosu/te/depthwise.py | 2 +-
.../contrib/ethosu/te/{depthwise.py => pooling.py} | 90 ++++------
.../relay/backend/contrib/ethosu/tir/compiler.py | 2 +-
.../tvm/relay/backend/contrib/ethosu/tir/passes.py | 2 +
.../ethosu/tir/{depthwise.py => pooling.py} | 62 ++-----
.../backend/contrib/ethosu/tir_to_cs_translator.py | 112 ++++++++----
python/tvm/relay/op/contrib/ethosu.py | 139 ++++++++++++++-
src/relay/op/contrib/ethosu/convolution.cc | 6 +-
src/relay/op/contrib/ethosu/depthwise.cc | 4 +-
src/relay/op/contrib/ethosu/pooling.cc | 195 +++++++++++++++++++++
tests/python/contrib/test_ethosu/infra.py | 49 +++++-
tests/python/contrib/test_ethosu/test_codegen.py | 89 ++++++++++
tests/python/contrib/test_ethosu/test_legalize.py | 101 ++++++++++-
.../contrib/test_ethosu/test_replace_pooling.py | 158 +++++++++++++++++
.../test_ethosu/test_tir_to_cs_translator.py | 79 ++++++++-
.../contrib/test_ethosu/test_type_inference.py | 91 +++++++++-
tests/python/driver/tvmc/test_compiler.py | 4 +-
24 files changed, 1211 insertions(+), 237 deletions(-)
copy python/tvm/relay/backend/contrib/ethosu/op/{depthwise.py => pooling.py} (58%)
copy python/tvm/relay/backend/contrib/ethosu/te/{depthwise.py => pooling.py} (59%)
copy python/tvm/relay/backend/contrib/ethosu/tir/{depthwise.py => pooling.py} (61%)
create mode 100644 src/relay/op/contrib/ethosu/pooling.cc
create mode 100644 tests/python/contrib/test_ethosu/test_replace_pooling.py