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

Remove MmaOpDetails::input_layout and getInputLayout #3322

Merged
merged 1 commit into from
Oct 31, 2024

Conversation

jacobhinkle
Copy link
Collaborator

There is no reason for us to check the Mma layout anymore when defining an MmaOp, since that is all handled in the scheduler now. I also added a test where a new batch dimension is broadcasted before defining the MmaOp.

Fixes #2273.

There is no reason for us to check the Mma layout anymore when defining
an MmaOp, since that is all handled in the scheduler now. I also added a
test where a new batch dimension is broadcasted before defining the
MmaOp.

Fixes #2273.
@jacobhinkle
Copy link
Collaborator Author

!build

@jacobhinkle jacobhinkle merged commit abdc3e1 into main Oct 31, 2024
18 of 19 checks passed
@jacobhinkle jacobhinkle deleted the remove_mma_layout_details branch October 31, 2024 18:16
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.

Cannot define an MmaOp where batch dimension is Broadcast
2 participants