bgawrych opened a new pull request #18320: URL: https://github.com/apache/incubator-mxnet/pull/18320
## Description ## Improve log_softmax op performance by using DNNL support **Native implementation:** | shape | time (ms/iter) | |---|---| | (5, 512, 512) | 5.3858582973480225 | | (5, 512, 1536) | 18.289981842041016 | | (5, 512, 2048) | 23.175915956497192 | | (5, 2048, 512) | 24.09080457687378 | | (4, 512, 512) | 4.271230220794678 | **MKLDNN implementation:** | shape | time (ms/iter) | |---|---| | (5, 512, 512) | 1.0982356071472168 | | (5, 512, 1536) | 3.9154343605041504 | | (5, 512, 2048) | 5.493232250213623 | | (5, 2048, 512) | 6.6350884437561035 | | (4, 512, 512) | 0.8532240390777588 | ## Checklist ## ### Essentials ### - [x] Changes are complete (i.e. I finished coding on this PR) - [x] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change ## Comments ## - Tests have increased tolerance - the reason of this is that DNNL sometimes have significantly different value of single element in array ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org
