This is an automated email from the ASF dual-hosted git repository.
lunderberg pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 9fb191159f [Unity] Commutative pattern match based on relax.Expr op
(#15494)
add 1a1af0c700 [Unity][NN] Allow nn.Module to generate subroutines (#15453)
No new revisions were added by this update.
Summary of changes:
python/tvm/relax/block_builder.py | 98 +++++++----
python/tvm/relax/expr.py | 8 +
python/tvm/relax/frontend/nn/__init__.py | 1 +
python/tvm/relax/frontend/nn/core.py | 14 +-
python/tvm/relax/frontend/nn/subroutine.py | 179 +++++++++++++++++++++
python/tvm/relax/testing/nn.py | 14 +-
python/tvm/relax/testing/relay_translator.py | 2 +-
tests/python/relax/test_frontend_nn_subroutines.py | 100 ++++++++++++
tests/python/relax/test_testing_nn.py | 62 ++++++-
tests/python/relax/test_tvmscript_ir_builder.py | 82 ++++++++++
10 files changed, 522 insertions(+), 38 deletions(-)
create mode 100644 python/tvm/relax/frontend/nn/subroutine.py
create mode 100644 tests/python/relax/test_frontend_nn_subroutines.py
