-
Notifications
You must be signed in to change notification settings - Fork 3.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ONNX] Support ScatterElements with reduction (#13894)
* add ScatterElements converter to ONNX front-end * native front-end for ScatterElements was implemented * update ScatterElements in ONNX high-level front-end * update comments * register ScatterElementsAttrs * register scatter elements strategy * implement generic scatter elements in topi * fix min-max redefinition * fix IntImm conversion and update scatter element implementation * fix parallel approach * CI tests for scatter elements were added * small update of description * sphinx issue was fixed * fix scatter deprecation in the CI test * fix * fix scatter version support * fix negative indices * add scatter elements strategy for cuda, gpu * update assert comment, update check of negative indices, hide tests for 18 version * fixes * extend error log for convenient analysis * lint fix * fix * sync dtypes * update cpu tir for scatter elements by scan example * scatter elements was basically implemented for topi/cuda * fix cpu scatter elements * fix gpu scatter elements * fix * small update * transfer indices check out of general loop * trancsfer ranges and strides calculation to gpu device * fixes * fix axis * clean code * fix after review * fix lint --------- Co-authored-by: Valery Chernov <[email protected]>
- Loading branch information
Showing
14 changed files
with
601 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.