Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cpu: aarch64: softmax: fall back to ref if stag != dtag #1823

Merged
merged 1 commit into from
Mar 12, 2024

Conversation

jondea
Copy link
Contributor

@jondea jondea commented Mar 6, 2024

acl_softmax would return incorrect answer if stag did not match dtag, for example

./benchdnn -v5 --softmax --stag=axb --dtag=abx 2x19x16x64

This patch fixes that by falling back to ref if stag != dtag.

Checklist

General

  • Do all unit and benchdnn tests (make test and make test_benchdnn_*) pass locally for each commit?
  • Have you formatted the code using clang-format?

Bug fixes

  • Have you included information on how to reproduce the issue (either in a github issue or in this PR)?
  • Have you added relevant regression tests? Not adding new tests because there is no new functionality, just restricting existing functionality.

`acl_softmax` would return incorrect answer if stag did not match dtag,
for example
```sh
./benchdnn -v5 --softmax --stag=axb --dtag=abx 2x19x16x64
```
This patch fixes that by falling back to ref if stag != dtag.
@vpirogov vpirogov added this to the v3.5 milestone Mar 7, 2024
@mgouicem mgouicem merged commit d3886c8 into oneapi-src:main Mar 12, 2024
10 checks passed
@vpirogov vpirogov added the platform:cpu-aarch64 Codeowner: @oneapi-src/onednn-cpu-aarch64 label May 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
platform:cpu-aarch64 Codeowner: @oneapi-src/onednn-cpu-aarch64
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants