This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 58a43c8724 [MetaSchedule] Fix a multilevel tiling error on dynamic
relax workload (#17465)
add c6a5b78690 [Relax] Enhance Relax op and ONNX frontend (#17462)
add 80250411e7 [Relax][MetaSchedule] Support CPU weight prepack (#17445)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/manipulate.h | 11 +
include/tvm/relax/transform.h | 21 ++
python/tvm/relax/frontend/nn/__init__.py | 2 +
python/tvm/relax/frontend/onnx/onnx_frontend.py | 66 ++++-
python/tvm/relax/op/__init__.py | 5 +
python/tvm/relax/op/binary.py | 26 ++
python/tvm/relax/op/create.py | 68 +++++
python/tvm/relax/op/manipulate.py | 44 +++
python/tvm/relax/pipeline.py | 50 +++-
python/tvm/relax/transform/__init__.py | 2 +
python/tvm/relax/transform/legalize_ops/binary.py | 3 +-
python/tvm/relax/transform/legalize_ops/create.py | 30 ++
.../tvm/relax/transform/legalize_ops/manipulate.py | 19 ++
python/tvm/relax/transform/transform.py | 29 ++
python/tvm/script/ir_builder/relax/ir.py | 10 +
python/tvm/topi/tensor.py | 35 ++-
src/meta_schedule/postproc/rewrite_layout.cc | 8 +-
src/relax/op/distributed/binary.cc | 2 +
src/relax/op/tensor/binary.cc | 2 +
src/relax/op/tensor/binary.h | 6 +
src/relax/op/tensor/create.cc | 84 ++++++
src/relax/op/tensor/create.h | 40 ++-
src/relax/op/tensor/manipulate.cc | 75 +++++
src/relax/op/tensor/manipulate.h | 12 +
.../transform/attach_attr_layout_free_buffers.cc | 113 +++++++
.../transform/split_layout_rewrite_preproc.cc | 327 +++++++++++++++++++++
.../test_meta_schedule_postproc_rewrite_layout.py | 3 +-
tests/python/relax/test_frontend_onnx.py | 26 +-
tests/python/relax/test_op_create.py | 58 ++++
tests/python/relax/test_op_manipulate.py | 52 ++++
...st_transform_attach_attr_layout_free_buffers.py | 311 ++++++++++++++++++++
.../test_transform_split_layout_rewrite_preproc.py | 220 ++++++++++++++
32 files changed, 1740 insertions(+), 20 deletions(-)
create mode 100644 src/relax/transform/attach_attr_layout_free_buffers.cc
create mode 100644 src/relax/transform/split_layout_rewrite_preproc.cc
create mode 100644
tests/python/relax/test_transform_attach_attr_layout_free_buffers.py
create mode 100644
tests/python/relax/test_transform_split_layout_rewrite_preproc.py