onnxruntime
Update ScatterElements to Support Opset 13, 15, 18
#19198
Merged
Overview
Commits
17
Changes
View On GitHub

Update ScatterElements to Support Opset 13, 15, 18 #19198

wschin merged 17 commits into main from wechi/scatter-elements
wschin
wschin wschin requested a review from xadupre xadupre 2 years ago
xadupre
xadupre commented on 2024-01-22
xadupre
xadupre commented on 2024-01-22
wschin Register ScatterElements for opset > 11
f3b33289
wschin Fix atomic
1e9a0a72
wschin wschin force pushed from df5b2e2d to 1e9a0a72 2 years ago
wschin documents
5274efe1
wschin Implement int8_t using template function
469245e0
wschin Rename a type
ce51bc62
wschin General impl for atomic binary ops
9ed1c5e7
wschin Changes:
9a59ff4c
wschin Implement for old cuda
42eac4f2
wschin Only keep one impl
736391af
wschin Copy change to ROCM
c8c97a83
wschin Try fix support version of atomicCAS
a37393e3
wschin exclude openvino & tensorrt for new tests
d456ee02
wschin define ScatterElements in ROCM
2970dc8d
wschin wschin marked this pull request as ready for review 2 years ago
wschin fix copy-and-paste
c77f2363
wschin ROCm doesn't support 16-bit atomicCAS. Let's fallback.
53e07672
wschin Hack to add doc change since doc generation code doesn't work on my end.
55cc23ad
wschin Fix a test
1b038805
wschin wschin force pushed from ede11d86 to 1b038805 2 years ago
prathikr
prathikr approved these changes on 2024-01-30
wschin wschin merged ffc3431a into main 2 years ago
wschin wschin deleted the wechi/scatter-elements branch 2 years ago
anjandeepsahni

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum