Skip to content

Annotate scaled matmul dispatch #22770

@Yu-Zhewen

Description

@Yu-Zhewen

Currently, only regular matmul is handled, and scaled matmul operations are instead classified as “reduction” dispatches.

} 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
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions