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 49564f7a2c Add masked_fill_.scalar, logical_not.default in Exported
Program frontend (#17909)
add 0c0cdd8958 Add op support for new_zeros op in Exported Program and fx
graph frontend (#17911)
add fd7fecc95a [Relax][PyTorch] Add support for eye op in fx graph (#17908)
add 85e368952d Fix off-by-one error in the type index range check within
Object::IsInstance()
add cbba0e38d3 Make slot end calculation more readable
add 4955f3b465 [Relax][PyTorch] Add Pixel Shuffle Op Support for Exported
Program and FX graph (#17886)
add c211e482be [Triton] Support latest `triton.compile` interface (#17913)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/nn.h | 9 +++
include/tvm/runtime/object.h | 2 +-
.../frontend/torch/base_fx_graph_translator.py | 28 ++++++++
.../frontend/torch/exported_program_translator.py | 2 +
python/tvm/relax/frontend/torch/fx_translator.py | 11 +++
python/tvm/relax/op/nn/__init__.py | 1 +
python/tvm/relax/op/nn/nn.py | 33 +++++++++
python/tvm/relax/transform/legalize_ops/nn.py | 6 ++
python/tvm/script/ir_builder/tir/triton.py | 23 +++++--
python/tvm/topi/nn/pixel_shuffle.py | 75 ++++++++++++++++++++
src/relax/op/nn/nn.cc | 70 +++++++++++++++++++
src/relax/op/nn/nn.h | 3 +
.../relax/test_frontend_from_exported_program.py | 59 ++++++++++++++++
tests/python/relax/test_frontend_from_fx.py | 79 ++++++++++++++++++++++
tests/python/relax/test_op_nn.py | 20 ++++++
15 files changed, 416 insertions(+), 5 deletions(-)
create mode 100644 python/tvm/topi/nn/pixel_shuffle.py
