-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Add oneDNN support for reduce operators #20669
Conversation
Hey @bgawrych , Thanks for submitting the PR
CI supported jobs: [sanity, miscellaneous, website, centos-cpu, clang, centos-gpu, windows-gpu, windows-cpu, unix-cpu, unix-gpu, edge] Note: |
3384558
to
e8bb0d2
Compare
e8bb0d2
to
f7bc409
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Overall looks good. Only some cosmetic changes.
4e2fadb
to
fd890eb
Compare
@mxnet-bot run ci [miscellaneous, windows-gpu] |
Jenkins CI successfully triggered : [miscellaneous, windows-gpu] |
Perf results on CLX8280 (28 cores): Script:
|
Merging, @szha if there is any procedure of merging own commits let me know |
Description
Add oneDNN support for reduction operators: sum and mean, for both nd and np interfaces