This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch unity-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
discard 2a0540a69a Merge branch 'apache-upstream-main' into
apache-upstream-unity
add d85ea256fa [Unity][TIR][Unittest] Fix failed 0-rank rfactor test
(#15846)
add 374b03f7e9 [Unity][Doc] Fix Python Docstring for Sphinx Build (#15848)
add 19c378dbe6 [Unity] Avoid Using Wildcard Imports (#15850)
add dd21ed37a8 [Unity][Relax]fix behaviors for importing torch.cat/concat
(#15845)
add 233ac7aee0 [Unity] Avoid trivial var-to-var bindings in
CanonicalizeBindings (#15840)
add 407e20d74a [Unity][Bugfix] Check for trivial re-bindings in MatchCast
nodes (#15854)
add 88a08ae47a [Disco] Loading-time sharding support (#15826)
add d618112932 [Unity][CUTLASS] Fix hardcoded epsilon parameter in RMS
norm (#15856)
add 8dda1cb22b [Unity][BYOC] Support attention over batched sequences of
variable sequence lengths (#15837)
add 4735a5b8e5 [UNITY] Handle pattern with remove_pad (#15843)
add 063cd7fb14 [Unity][Doc] Fix Relax part fix_docstring (#15860)
new 11c73a2ea6 Merge remote-tracking branch 'apache-upstream/main' into
unity-staging
This update added new revisions after undoing existing revisions.
That is to say, some revisions that were in the old version of the
branch are not in the new version. This situation occurs
when a user --force pushes a change and generates a repository
containing something like this:
* -- * -- B -- O -- O -- O (2a0540a69a)
\
N -- N -- N refs/heads/unity-staging (11c73a2ea6)
You should already have received notification emails for all of the O
revisions, and so the following emails describe only the N revisions
from the common base, B.
Any revisions marked "omit" are not gone; other references still
refer to them. Any revisions marked "discard" are gone forever.
The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
3rdparty/cutlass | 2 +-
python/tvm/contrib/cutlass/attention_operation.py | 11 ++
python/tvm/contrib/cutlass/build.py | 55 +++---
python/tvm/contrib/cutlass/gen_tensor_op.py | 12 +-
python/tvm/contrib/cutlass/rms_norm_operation.py | 2 +-
python/tvm/relax/analysis/__init__.py | 27 ++-
python/tvm/relax/analysis/analysis.py | 4 +-
python/tvm/relax/backend/contrib/cutlass.py | 4 +
python/tvm/relax/backend/patterns.py | 23 ++-
python/tvm/relax/block_builder.py | 2 +-
python/tvm/relax/distributed/transform/__init__.py | 3 +-
python/tvm/relax/expr.py | 4 +-
python/tvm/relax/expr_functor.py | 8 +
python/tvm/relax/frontend/nn/op.py | 1 +
python/tvm/relax/frontend/torch/fx_translator.py | 4 +-
python/tvm/relax/op/__init__.py | 142 ++++++++++++---
python/tvm/relax/op/base.py | 6 +-
python/tvm/relax/op/builtin/__init__.py | 3 +-
python/tvm/relax/op/ccl/__init__.py | 3 +-
python/tvm/relax/op/distributed/__init__.py | 3 +-
python/tvm/relax/op/grad/__init__.py | 1 -
python/tvm/relax/op/image/__init__.py | 3 +-
python/tvm/relax/op/manipulate.py | 9 +
python/tvm/relax/op/memory/__init__.py | 3 +-
python/tvm/relax/op/nn/__init__.py | 27 ++-
python/tvm/relax/op/nn/nn.py | 121 ++++++++++++-
python/tvm/relax/op/vm/__init__.py | 3 +-
python/tvm/relax/struct_info.py | 2 +-
python/tvm/relax/transform/__init__.py | 58 +++++-
python/tvm/relax/transform/legalize_ops/nn.py | 5 +
.../relax/transform/optimize_layout_transform.py | 23 ++-
python/tvm/relax/transform/transform.py | 47 ++---
python/tvm/relax/vm_build.py | 1 +
python/tvm/runtime/disco/process_pool.py | 1 -
python/tvm/runtime/relax_vm.py | 21 ++-
python/tvm/script/ir_builder/relax/__init__.py | 4 +-
python/tvm/te/operation.py | 12 +-
src/relax/op/nn/attention.cc | 31 +++-
src/relax/transform/canonicalize_bindings.cc | 49 +++--
src/relax/transform/kill_after_last_use.cc | 17 ++
src/runtime/contrib/thrust/thrust.cu | 8 +-
src/runtime/disco/loader.cc | 198 ++++++++++-----------
src/runtime/relax_vm/ndarray_cache_support.cc | 85 ++++++---
src/runtime/relax_vm/ndarray_cache_support.h | 19 +-
tests/python/disco/test_loader.py | 193 ++++++++++++++++----
tests/python/relax/test_codegen_cutlass.py | 106 ++++++++++-
tests/python/relax/test_frontend_from_fx.py | 49 +++++
tests/python/relax/test_kill_after_last_use.py | 51 ++++++
.../python/relax/test_optimize_layout_transform.py | 143 +++++++++++++++
.../relax/test_transform_canonicalize_bindings.py | 41 +----
tests/python/unittest/test_tir_schedule_rfactor.py | 2 +-
51 files changed, 1291 insertions(+), 361 deletions(-)
