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

Flip the order in which the GEMM tiles are computed, i.e. instead of traversing the tiles in n direction first, traverse them in m direction, and then in n direction. #7513

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

copybara-service[bot]
Copy link
Contributor

Flip the order in which the GEMM tiles are computed, i.e. instead of traversing the tiles in n direction first, traverse them in m direction, and then in n direction.

Some minor improvements due to better cache re-use across tiles where the tile's ncxk weights fit in the lowest-level cache.

…traversing the tiles in `n` direction first, traverse them in `m` direction, and then in `n` direction.

Some minor improvements due to better cache re-use across tiles where the tile's `nc`x`k` weights fit in the lowest-level cache.

PiperOrigin-RevId: 690625625
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant