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/17 09:51:46 UTC
[tvm] branch main updated (95ff358 -> ee23013)
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 95ff358 [microNPU] Allow constants to be given as input to an operator (#9515)
add ee23013 [4/10] Code generation for Conv2D via CMSIS-NN (#9331)
No new revisions were added by this update.
Summary of changes:
python/tvm/micro/model_library_format.py | 12 +-
python/tvm/relay/op/contrib/cmsisnn.py | 83 ++++--
src/relay/backend/aot_executor_codegen.cc | 28 +-
.../backend/contrib/cmsisnn/extract_constants.cc | 168 +++++++++++
.../backend/contrib/cmsisnn/generate_constants.cc | 244 ++++++++++++++++
src/relay/backend/contrib/cmsisnn/relay_to_tir.cc | 174 +++++++++++-
.../backend/contrib/cmsisnn/tir_to_runtime.cc | 164 +++++++++++
src/relay/backend/graph_executor_codegen.cc | 2 +-
src/relay/backend/interpreter.cc | 2 +-
src/relay/backend/te_compiler.cc | 44 ++-
src/relay/backend/te_compiler.h | 10 +-
src/relay/backend/utils.h | 2 +-
src/relay/backend/vm/compiler.cc | 3 +-
src/target/source/codegen_c_host.h | 2 +-
.../python/contrib/test_cmsisnn/test_binary_ops.py | 1 +
tests/python/contrib/test_cmsisnn/test_conv2d.py | 316 +++++++++++++++++++++
.../contrib/test_cmsisnn/test_extract_constants.py | 179 ++++++++++++
.../test_cmsisnn/test_generate_constants.py | 229 +++++++++++++++
tests/python/contrib/test_cmsisnn/test_networks.py | 12 +-
tests/python/contrib/test_cmsisnn/test_softmax.py | 6 +-
tests/python/contrib/test_cmsisnn/utils.py | 127 ++++++++-
21 files changed, 1719 insertions(+), 89 deletions(-)
create mode 100644 src/relay/backend/contrib/cmsisnn/extract_constants.cc
create mode 100644 src/relay/backend/contrib/cmsisnn/generate_constants.cc
create mode 100644 tests/python/contrib/test_cmsisnn/test_conv2d.py
create mode 100644 tests/python/contrib/test_cmsisnn/test_extract_constants.py
create mode 100644 tests/python/contrib/test_cmsisnn/test_generate_constants.py