This is an automated email from the ASF dual-hosted git repository.
andrewzhaoluo pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 0dd3d4aab6 [ACL][TESTING] Use pytest.mark.parametrize in ACL conv2d
tests (#14011)
add 0c965f442a [ONNX] Support ScatterElements with reduction (#13894)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/transform.h | 12 +++
python/tvm/relay/frontend/onnx.py | 52 ++++++++-
python/tvm/relay/op/_transform.py | 10 ++
python/tvm/relay/op/op_attrs.py | 5 +
python/tvm/relay/op/strategy/cuda.py | 14 +++
python/tvm/relay/op/strategy/generic.py | 22 ++++
python/tvm/relay/op/transform.py | 35 ++++++
python/tvm/topi/__init__.py | 1 +
python/tvm/topi/cuda/__init__.py | 1 +
python/tvm/topi/cuda/scatter_elements.py | 167 +++++++++++++++++++++++++++++
python/tvm/topi/scatter_elements.py | 150 ++++++++++++++++++++++++++
src/relay/op/tensor/transform.cc | 56 ++++++++++
src/tir/ir/expr.cc | 4 +-
tests/python/frontend/onnx/test_forward.py | 80 +++++++++++++-
14 files changed, 601 insertions(+), 8 deletions(-)
create mode 100644 python/tvm/topi/cuda/scatter_elements.py
create mode 100644 python/tvm/topi/scatter_elements.py
