You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by wu...@apache.org on 2022/09/27 23:49:38 UTC

[tvm] branch main updated (c89a8baeeb -> 178f82dc48)

This is an automated email from the ASF dual-hosted git repository.

wuwei pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


    from c89a8baeeb [usmp] Also remap VarNode to USMP-allocated buffer (#12880)
     add 178f82dc48 [TOPI] Implement Einsum with reduction axes (#12913)

No new revisions were added by this update.

Summary of changes:
 include/tvm/topi/einsum.h                    | 889 +--------------------------
 src/relay/op/tensor/math.cc                  |   2 +-
 src/topi/einsum.cc                           | 353 +++++++++++
 src/topi/transform.cc                        |   4 -
 tests/python/topi/python/test_topi_einsum.py |  36 +-
 5 files changed, 397 insertions(+), 887 deletions(-)
 create mode 100644 src/topi/einsum.cc