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.

Sign up for GitHub

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

Jump to bottom

[QST] Row Major epilogue operands in SM90 GEMMs #1265

Closed
alihassanijr opened this issue Dec 12, 2023 · 2 comments
Closed

[QST] Row Major epilogue operands in SM90 GEMMs #1265

alihassanijr opened this issue Dec 12, 2023 · 2 comments
Labels
? - Needs Triage question Question

Comments

@alihassanijr
Copy link
Contributor

alihassanijr commented Dec 12, 2023 •
edited
Loading

Can SM90 WGMMAs built with collective builders support row-major layouts for epilogue operands (C/D)?

cutlass_library is only emitting kernel instantiations with column-major epilogues:

Thank you.
@thakkarV
Copy link
Collaborator

Yes. Row major output tensor layouts are fully supported by both TMA and the direct store epilogues.

@alihassanijr
Copy link
Contributor Author

Thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
? - Needs Triage question Question
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@thakkarV @alihassanijr