This is an automated email from the ASF dual-hosted git repository.
ruihangl pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 876ed385ce [Unity][Op] Symbolic shape support of take grad (#14559)
add fadf144f27 [Unity][Op] add einsum and flip in Relax (#14545)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/linear_algebra.h | 9 ++
include/tvm/relax/attrs/manipulate.h | 21 ++---
include/tvm/relax/attrs/statistical.h | 15 ++++
python/tvm/relax/op/_op_gradient.py | 3 +-
python/tvm/relax/op/linear_algebra.py | 24 +++++-
python/tvm/relax/op/manipulate.py | 43 +++-------
python/tvm/relax/op/op_attrs.py | 10 +++
python/tvm/relax/op/statistical.py | 51 ++++++++++++
.../relax/transform/legalize_ops/linear_algebra.py | 24 +++++-
.../tvm/relax/transform/legalize_ops/manipulate.py | 6 +-
.../relax/transform/legalize_ops/statistical.py | 5 ++
python/tvm/script/ir_builder/relax/ir.py | 4 +
src/relax/op/op_common.cc | 25 ++++++
src/relax/op/op_common.h | 11 +++
src/relax/op/tensor/linear_algebra.cc | 65 +++++++++++++++
src/relax/op/tensor/linear_algebra.h | 8 ++
src/relax/op/tensor/manipulate.cc | 82 +++++-------------
src/relax/op/tensor/manipulate.h | 10 +--
src/relax/op/tensor/statistical.cc | 47 +++++++++++
src/relax/op/tensor/statistical.h | 12 +++
src/topi/einsum.cc | 5 +-
tests/python/relax/test_op_linear_algebra.py | 83 ++++++++++++++++++
tests/python/relax/test_op_manipulate.py | 61 ++++----------
tests/python/relax/test_op_statistical.py | 62 ++++++++++++++
..._transform_legalize_ops_index_linear_algebra.py | 97 +++++++++++++++++++++-
.../test_transform_legalize_ops_manipulate.py | 91 ++++++++++----------
...st_transform_legalize_ops_search_statistical.py | 73 ++++++++++++++++
.../test_tvmscript_parser_op_linear_algebra.py | 16 ++++
.../relax/test_tvmscript_parser_op_manipulate.py | 6 +-
.../relax/test_tvmscript_parser_op_statistical.py | 15 ++++
30 files changed, 769 insertions(+), 215 deletions(-)
