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 89b91e2b11 [KVCache] Partial layers support (#17192)
add 9a07870b2e [CLML][CI] Fix for few clml regression issues (#17117)
add ae1be53d6d [Disco] Cross-group and p2p send/receive primitives (#17191)
add 9f0f301c6f [TIR][Analyzer] Simplify `x==x` expressions for all dtypes
(#17158)
add cc8afdb0e3 Add support for `torch.nn.functional.max_pool2d` (#17189)
add 7bd738a00b [Relax] Implement Rewriter class for pattern-rewrite
(#17149)
add 6704175fc7 Pass to eliminate redundant branch and overcompute (#17170)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/block_builder.h | 35 +-
include/tvm/relax/expr_functor.h | 21 +-
include/tvm/runtime/disco/builtin.h | 24 +
include/tvm/script/ir_builder/relax/frame.h | 1 +
include/tvm/tir/transform.h | 8 +
python/tvm/relax/dpl/__init__.py | 8 +-
python/tvm/relax/dpl/rewrite.py | 186 ++-
python/tvm/relax/frontend/nn/core.py | 6 +-
python/tvm/relax/frontend/torch/fx_translator.py | 1 +
python/tvm/relay/op/contrib/clml.py | 8 +-
python/tvm/script/ir_builder/relax/ir.py | 48 +-
python/tvm/script/parser/core/utils.py | 14 +-
python/tvm/tir/transform/transform.py | 13 +
src/arith/rewrite_simplify.cc | 21 +-
src/relax/ir/block_builder.cc | 95 +-
src/relax/ir/dataflow_block_rewriter.cc | 452 ++++++
src/relax/ir/dataflow_expr_rewriter.cc | 1079 ++++++++++++++
src/relax/ir/dataflow_matcher.cc | 669 +--------
...{dataflow_matcher_impl.h => dataflow_matcher.h} | 15 +-
src/relax/ir/dataflow_rewriter.h | 182 +++
src/relax/ir/expr.cc | 42 +-
src/relax/ir/expr_functor.cc | 54 +-
src/relax/transform/canonicalize_bindings.cc | 142 +-
src/relax/transform/utils.h | 2 +-
src/relax/utils.cc | 16 +-
src/runtime/disco/builtin.cc | 16 +
src/runtime/disco/nccl/nccl.cc | 86 ++
src/script/ir_builder/relax/frame.cc | 7 +-
src/script/ir_builder/relax/ir.cc | 10 +-
.../transforms/using_assume_to_reduce_branches.cc | 394 +++++
tests/python/arith/test_arith_rewrite_simplify.py | 36 +
tests/python/arith/test_arith_simplify.py | 29 +
tests/python/disco/test_ccl.py | 40 +-
tests/python/relax/test_dataflow_rewriter.py | 1512 ++++++++++++++++++++
...eliminate_pad_branch_using_buffer_assumption.py | 648 +++++++++
tests/python/relax/test_frontend_from_fx.py | 8 +
.../relax/test_transform_canonicalize_bindings.py | 255 +++-
.../test_transform_legalize_ops_manipulate.py | 2 +-
tests/python/relax/test_tvmscript_parser.py | 46 +
tests/scripts/setup-adreno-env.sh | 1 +
tests/scripts/task_python_adreno.sh | 3 +-
41 files changed, 5473 insertions(+), 762 deletions(-)
create mode 100644 src/relax/ir/dataflow_block_rewriter.cc
create mode 100644 src/relax/ir/dataflow_expr_rewriter.cc
rename src/relax/ir/{dataflow_matcher_impl.h => dataflow_matcher.h} (91%)
create mode 100644 src/relax/ir/dataflow_rewriter.h
create mode 100644 src/tir/transforms/using_assume_to_reduce_branches.cc
create mode 100644 tests/python/relax/test_dataflow_rewriter.py
create mode 100644
tests/python/relax/test_eliminate_pad_branch_using_buffer_assumption.py
