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: brgemm: Fix unimplemented conditions for brgemm #2148

Merged
merged 1 commit into from
Oct 4, 2024

Conversation

Ryo-not-rio
Copy link
Contributor

Description

Removing asserts in brgemm accidentally flipped the validation conditions for brgemm, causing brgemm to go to the gemm:jit implementation instead of brg:sve_256 implementation. This is a fix for that issue.

Bug fixes

The issue can be reproduced by running the ./tests/benchdnn/benchdnn --matmul 2048x479:479x1024. The observed behaviour is gemm:jit being used whereas this fix should cause brg:sve_256 to be run.

  • Have you included information on how to reproduce the issue (either in a github issue or in this PR)?

@theComputeKid
Copy link
Contributor

@Shreyas-fuj can you please have a look. Thanks.

@theComputeKid theComputeKid changed the title cpu:aarch64:brgemm: Fix unimplemented conditions for brgemm cpu: aarch64: brgemm: Fix unimplemented conditions for brgemm Oct 4, 2024
@Shreyas-fuj
Copy link
Contributor

@Shreyas-fuj can you please have a look. Thanks.

@theComputeKid, @Ryo-not-rio, I tested from my end it seems fine to me, thanks.

Change-Id: I3be1901c6209a9ace14eda073c90f4c4098d228d
@Ryo-not-rio Ryo-not-rio reopened this Oct 4, 2024
@spalicki spalicki merged commit cfe12d8 into oneapi-src:main Oct 4, 2024
24 of 26 checks passed
@github-actions github-actions bot added the platform:cpu-aarch64 Codeowner: @oneapi-src/onednn-cpu-aarch64 label Oct 8, 2024
@Radu2k Radu2k mentioned this pull request Dec 2, 2024
@vpirogov vpirogov added this to the v3.7 milestone Dec 11, 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.

6 participants