Skip to content

Commit 51db978

Browse files
GroverkssElias Joseph
authored andcommitted
[ExternalInterfaces] Fix LinalgExtFusionInterface registeration (#20244)
addExtension adds a dialect dependency on the dialect passed as an argument. If LinalgExtDialect is never registered, we would never register LinalgExtFusionInterface for linalg ops. Signed-off-by: Elias Joseph <[email protected]>
1 parent de28048 commit 51db978

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

compiler/src/iree/compiler/ExternalInterfaces/LinalgExtExternalModels.cpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,10 +123,7 @@ void registerOpsWithLinalgExtOpInterface(mlir::MLIRContext *context) {
123123
} // namespace
124124

125125
void registerLinalgExtExternalModels(DialectRegistry &registry) {
126-
registry.addExtension(+[](MLIRContext *ctx,
127-
IREE::LinalgExt::IREELinalgExtDialect *dialect) {
128-
ctx->loadDialect<mlir::linalg::LinalgDialect>();
129-
126+
registry.addExtension(+[](MLIRContext *ctx, linalg::LinalgDialect *dialect) {
130127
#define GET_OP_LIST
131128
registerOpsWithLinalgExtOpInterface<
132129
#include "mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc"

0 commit comments

Comments
 (0)