search

NVIDIA / Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
Other
271 stars 53 forks source link

Remove MmaOpDetails::input_layout and getInputLayout #3322

Closed jacobhinkle closed 3 weeks ago

jacobhinkle commented 3 weeks ago

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 commented 3 weeks ago

!build