Skip to content

Remove MmaOpDetails::input_layout and getInputLayout#3322

Merged
jacobhinkle merged 1 commit intomainfrom
remove_mma_layout_details
Oct 31, 2024
Merged

Remove MmaOpDetails::input_layout and getInputLayout#3322
jacobhinkle merged 1 commit intomainfrom
remove_mma_layout_details

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
@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