This is an automated email from the ASF dual-hosted git repository.
zhic pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from f5af03a [Relay] Fix dynamic case for Squeeze and Split (#6739)
add 3d624ec Scatter on Cuda (#6533)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/_transform.py | 2 +-
python/tvm/relay/op/strategy/cuda.py | 13 +
python/tvm/relay/op/strategy/generic.py | 23 +-
python/tvm/topi/cuda/__init__.py | 1 +
python/tvm/topi/cuda/scatter.py | 443 ++++++++++++++++++++++++++++++++
tests/python/relay/test_op_level3.py | 41 ++-
6 files changed, 515 insertions(+), 8 deletions(-)
create mode 100644 python/tvm/topi/cuda/scatter.py