-
Notifications
You must be signed in to change notification settings - Fork 802
Open
Description
Currently, only regular matmul is handled, and scaled matmul operations are instead classified as “reduction” dispatches.
iree/compiler/src/iree/compiler/Dialect/Flow/Transforms/AnnotateDispatches.cpp
Lines 296 to 305 in 39a15a7
| } else if (isMatmulLike(op)) { | |
| prefix = "matmul_like"; | |
| } else if (linalg::isaContractionOpInterface(op)) { | |
| prefix = "contract"; | |
| } else if (IREE::LinalgExt::isaHorizontallyFusedContraction(op)) { | |
| prefix = "horizontal_multi_contract"; | |
| } else if (succeeded(linalg::inferConvolutionDims(op))) { | |
| prefix = "conv"; | |
| } else if (op.getNumReductionLoops()) { | |
| prefix = "reduction"; |
Metadata
Metadata
Assignees
Labels
No labels