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