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

Handle large strides in matmul properly #2611

Merged
merged 1 commit into from
Feb 6, 2025

Conversation

densamoilov
Copy link
Contributor

Fixes MFDNN-13075.

This PR adds checks for large (> INT32_MAX) strides to brgemm-based matmul primitive so that such cases can be properly handled by returning an "unimplemented".
Adding support for large strides will require extending brgemm and copy routines which doesn't seem to be a good investment at this point because the shapes are not used in real world workloads for CPU.

@densamoilov densamoilov requested a review from a team as a code owner February 6, 2025 08:30
@github-actions github-actions bot added the platform:cpu-x64 Intel64/AMD64 processors. Codeowner: @oneapi-src/onednn-cpu-x64 label Feb 6, 2025
@densamoilov
Copy link
Contributor Author

make test

@densamoilov densamoilov changed the title cpu: x64: matmul: add check for large strides Handle large strides in matmul properly Feb 6, 2025
@densamoilov densamoilov merged commit 8985280 into main Feb 6, 2025
13 of 14 checks passed
@densamoilov densamoilov deleted the dsamoylo/main/matmul-skip-large-strides branch February 6, 2025 18:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
platform:cpu-x64 Intel64/AMD64 processors. Codeowner: @oneapi-src/onednn-cpu-x64
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants