Skip to content

Commit 26669bd

Browse files
nikalranikalra
committed
attach extension IDs
1 parent c653279 commit 26669bd

File tree

93 files changed

+567
-393
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

93 files changed

+567
-393
lines changed

‎flang/lib/Optimizer/Dialect/FIRDialect.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ void fir::FIROpsDialect::initialize() {
7474
// Register the FIRInlinerInterface to FIROpsDialect
7575
void fir::addFIRInlinerExtension(mlir::DialectRegistry &registry) {
7676
registry.addExtension(
77-
+[](mlir::MLIRContext *ctx, fir::FIROpsDialect *dialect) {
77+
"FIR_INLINER", +[](mlir::MLIRContext *ctx, fir::FIROpsDialect *dialect) {
7878
dialect->addInterface<FIRInlinerInterface>();
7979
});
8080
}
@@ -90,6 +90,7 @@ void fir::addFIRInlinerExtension(mlir::DialectRegistry &registry) {
9090
// when more sophisticated translation is required.
9191
void fir::addFIRToLLVMIRExtension(mlir::DialectRegistry &registry) {
9292
registry.addExtension(
93+
"FIR_LLVM_TRANSLATION",
9394
+[](mlir::MLIRContext *ctx, fir::FIROpsDialect *dialect) {
9495
dialect->addInterface<mlir::LLVMTranslationDialectInterface>();
9596
});

‎mlir/examples/transform/Ch2/lib/MyExtension.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,9 @@ class MyExtension
3636
// dialect definitions. List individual operations and dependent dialects
3737
// here.
3838
void init();
39+
40+
// Declare a unique ID for this extension.
41+
static constexpr llvm::StringRef extensionID = "CH2_TRANSFORM";
3942
};
4043

4144
void MyExtension::init() {

‎mlir/examples/transform/Ch3/lib/MyExtension.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@ class MyExtension
4242
// dialect definitions. List individual operations and dependent dialects
4343
// here.
4444
void init();
45+
46+
// Declare a unique ID for this extension.
47+
static constexpr llvm::StringRef extensionID = "CH3_TRANSFORM";
4548
};
4649

4750
void MyExtension::init() {

‎mlir/examples/transform/Ch4/lib/MyExtension.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@ class MyExtension
3838
// dialect definitions. List individual operations and dependent dialects
3939
// here.
4040
void init();
41+
42+
// Declare a unique ID for this extension.
43+
static constexpr llvm::StringRef extensionID = "CH4_TRANSFORM";
4144
};
4245

4346
void MyExtension::init() {

‎mlir/lib/Conversion/ArithToLLVM/ArithToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -510,9 +510,10 @@ struct ArithToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
510510

511511
void mlir::arith::registerConvertArithToLLVMInterface(
512512
DialectRegistry &registry) {
513-
registry.addExtension(+[](MLIRContext *ctx, arith::ArithDialect *dialect) {
514-
dialect->addInterfaces<ArithToLLVMDialectInterface>();
515-
});
513+
registry.addExtension(
514+
"ARITH_TO_LLVM", +[](MLIRContext *ctx, arith::ArithDialect *dialect) {
515+
dialect->addInterfaces<ArithToLLVMDialectInterface>();
516+
});
516517
}
517518

518519
//===----------------------------------------------------------------------===//

‎mlir/lib/Conversion/ComplexToLLVM/ComplexToLLVM.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -385,6 +385,7 @@ struct ComplexToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
385385

386386
void mlir::registerConvertComplexToLLVMInterface(DialectRegistry &registry) {
387387
registry.addExtension(
388+
"COMPLEX_TO_LLVM",
388389
+[](MLIRContext *ctx, complex::ComplexDialect *dialect) {
389390
dialect->addInterfaces<ComplexToLLVMDialectInterface>();
390391
});

‎mlir/lib/Conversion/ControlFlowToLLVM/ControlFlowToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,8 @@ struct ControlFlowToLLVMDialectInterface
273273

274274
void mlir::cf::registerConvertControlFlowToLLVMInterface(
275275
DialectRegistry &registry) {
276-
registry.addExtension(+[](MLIRContext *ctx, cf::ControlFlowDialect *dialect) {
277-
dialect->addInterfaces<ControlFlowToLLVMDialectInterface>();
278-
});
276+
registry.addExtension(
277+
"CF_TO_LLVM", +[](MLIRContext *ctx, cf::ControlFlowDialect *dialect) {
278+
dialect->addInterfaces<ControlFlowToLLVMDialectInterface>();
279+
});
279280
}

‎mlir/lib/Conversion/ConvertToLLVM/ConvertToLLVMPass.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "mlir/Pass/Pass.h"
1616
#include "mlir/Rewrite/FrozenRewritePatternSet.h"
1717
#include "mlir/Transforms/DialectConversion.h"
18+
#include "llvm/ADT/StringRef.h"
1819
#include <memory>
1920

2021
#define DEBUG_TYPE "convert-to-llvm"
@@ -54,6 +55,10 @@ class LoadDependentDialectExtension : public DialectExtensionBase {
5455
std::unique_ptr<DialectExtensionBase> clone() const final {
5556
return std::make_unique<LoadDependentDialectExtension>(*this);
5657
}
58+
59+
/// Unique ID
60+
constexpr static llvm::StringRef extensionID =
61+
"CONVERT_TO_LLVM_LOAD_DEPENDENT";
5762
};
5863

5964
/// This is a generic pass to convert to LLVM, it uses the

‎mlir/lib/Conversion/FuncToLLVM/FuncToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -786,7 +786,8 @@ struct FuncToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
786786
} // namespace
787787

788788
void mlir::registerConvertFuncToLLVMInterface(DialectRegistry &registry) {
789-
registry.addExtension(+[](MLIRContext *ctx, func::FuncDialect *dialect) {
790-
dialect->addInterfaces<FuncToLLVMDialectInterface>();
791-
});
789+
registry.addExtension(
790+
"FUNC_TO_LLVM", +[](MLIRContext *ctx, func::FuncDialect *dialect) {
791+
dialect->addInterfaces<FuncToLLVMDialectInterface>();
792+
});
792793
}

‎mlir/lib/Conversion/IndexToLLVM/IndexToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,8 @@ struct IndexToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
392392

393393
void mlir::index::registerConvertIndexToLLVMInterface(
394394
DialectRegistry &registry) {
395-
registry.addExtension(+[](MLIRContext *ctx, index::IndexDialect *dialect) {
396-
dialect->addInterfaces<IndexToLLVMDialectInterface>();
397-
});
395+
registry.addExtension(
396+
"INDEX_TO_LLVM", +[](MLIRContext *ctx, index::IndexDialect *dialect) {
397+
dialect->addInterfaces<IndexToLLVMDialectInterface>();
398+
});
398399
}

‎mlir/lib/Conversion/MathToLLVM/MathToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,8 @@ struct MathToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
356356
} // namespace
357357

358358
void mlir::registerConvertMathToLLVMInterface(DialectRegistry &registry) {
359-
registry.addExtension(+[](MLIRContext *ctx, math::MathDialect *dialect) {
360-
dialect->addInterfaces<MathToLLVMDialectInterface>();
361-
});
359+
registry.addExtension(
360+
"MATH_TO_LLVM", +[](MLIRContext *ctx, math::MathDialect *dialect) {
361+
dialect->addInterfaces<MathToLLVMDialectInterface>();
362+
});
362363
}

‎mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1753,7 +1753,8 @@ struct MemRefToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
17531753
} // namespace
17541754

17551755
void mlir::registerConvertMemRefToLLVMInterface(DialectRegistry &registry) {
1756-
registry.addExtension(+[](MLIRContext *ctx, memref::MemRefDialect *dialect) {
1757-
dialect->addInterfaces<MemRefToLLVMDialectInterface>();
1758-
});
1756+
registry.addExtension(
1757+
"MEMREF_TO_LLVM", +[](MLIRContext *ctx, memref::MemRefDialect *dialect) {
1758+
dialect->addInterfaces<MemRefToLLVMDialectInterface>();
1759+
});
17591760
}

‎mlir/lib/Conversion/NVVMToLLVM/NVVMToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,8 @@ void mlir::populateNVVMToLLVMConversionPatterns(RewritePatternSet &patterns) {
113113
}
114114

115115
void mlir::registerConvertNVVMToLLVMInterface(DialectRegistry &registry) {
116-
registry.addExtension(+[](MLIRContext *ctx, NVVMDialect *dialect) {
117-
dialect->addInterfaces<NVVMToLLVMDialectInterface>();
118-
});
116+
registry.addExtension(
117+
"NVVM_TO_LLVM", +[](MLIRContext *ctx, NVVMDialect *dialect) {
118+
dialect->addInterfaces<NVVMToLLVMDialectInterface>();
119+
});
119120
}

‎mlir/lib/Conversion/UBToLLVM/UBToLLVM.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,8 @@ struct UBToLLVMDialectInterface : public ConvertToLLVMPatternInterface {
119119
} // namespace
120120

121121
void mlir::ub::registerConvertUBToLLVMInterface(DialectRegistry &registry) {
122-
registry.addExtension(+[](MLIRContext *ctx, ub::UBDialect *dialect) {
123-
dialect->addInterfaces<UBToLLVMDialectInterface>();
124-
});
122+
registry.addExtension(
123+
"UB_TO_LLVM", +[](MLIRContext *ctx, ub::UBDialect *dialect) {
124+
dialect->addInterfaces<UBToLLVMDialectInterface>();
125+
});
125126
}

‎mlir/lib/Dialect/Affine/IR/ValueBoundsOpInterfaceImpl.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,11 +96,12 @@ struct AffineMaxOpInterface
9696

9797
void mlir::affine::registerValueBoundsOpInterfaceExternalModels(
9898
DialectRegistry &registry) {
99-
registry.addExtension(+[](MLIRContext *ctx, AffineDialect *dialect) {
100-
AffineApplyOp::attachInterface<AffineApplyOpInterface>(*ctx);
101-
AffineMaxOp::attachInterface<AffineMaxOpInterface>(*ctx);
102-
AffineMinOp::attachInterface<AffineMinOpInterface>(*ctx);
103-
});
99+
registry.addExtension(
100+
"AFFINE_VALUE_BOUNDS", +[](MLIRContext *ctx, AffineDialect *dialect) {
101+
AffineApplyOp::attachInterface<AffineApplyOpInterface>(*ctx);
102+
AffineMaxOp::attachInterface<AffineMaxOpInterface>(*ctx);
103+
AffineMinOp::attachInterface<AffineMinOpInterface>(*ctx);
104+
});
104105
}
105106

106107
FailureOr<int64_t>

‎mlir/lib/Dialect/Affine/TransformOps/AffineTransformOps.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,8 @@ class AffineTransformDialectExtension
167167
#include "mlir/Dialect/Affine/TransformOps/AffineTransformOps.cpp.inc"
168168
>();
169169
}
170+
171+
static constexpr llvm::StringRef extensionID = "AFFINE_TRANSFORM";
170172
};
171173
} // namespace
172174

‎mlir/lib/Dialect/Arith/IR/ValueBoundsOpInterfaceImpl.cpp

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -150,11 +150,13 @@ struct SelectOpInterface
150150

151151
void mlir::arith::registerValueBoundsOpInterfaceExternalModels(
152152
DialectRegistry &registry) {
153-
registry.addExtension(+[](MLIRContext *ctx, arith::ArithDialect *dialect) {
154-
arith::AddIOp::attachInterface<arith::AddIOpInterface>(*ctx);
155-
arith::ConstantOp::attachInterface<arith::ConstantOpInterface>(*ctx);
156-
arith::SubIOp::attachInterface<arith::SubIOpInterface>(*ctx);
157-
arith::MulIOp::attachInterface<arith::MulIOpInterface>(*ctx);
158-
arith::SelectOp::attachInterface<arith::SelectOpInterface>(*ctx);
159-
});
153+
registry.addExtension(
154+
"ARITH_VALUE_BOUNDS",
155+
+[](MLIRContext *ctx, arith::ArithDialect *dialect) {
156+
arith::AddIOp::attachInterface<arith::AddIOpInterface>(*ctx);
157+
arith::ConstantOp::attachInterface<arith::ConstantOpInterface>(*ctx);
158+
arith::SubIOp::attachInterface<arith::SubIOpInterface>(*ctx);
159+
arith::MulIOp::attachInterface<arith::MulIOpInterface>(*ctx);
160+
arith::SelectOp::attachInterface<arith::SelectOpInterface>(*ctx);
161+
});
160162
}

‎mlir/lib/Dialect/Arith/Transforms/BufferDeallocationOpInterfaceImpl.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,8 @@ struct SelectOpInterface
7979

8080
void mlir::arith::registerBufferDeallocationOpInterfaceExternalModels(
8181
DialectRegistry &registry) {
82-
registry.addExtension(+[](MLIRContext *ctx, ArithDialect *dialect) {
83-
SelectOp::attachInterface<SelectOpInterface>(*ctx);
84-
});
82+
registry.addExtension(
83+
"ARITH_BUFFER_DEALLOC", +[](MLIRContext *ctx, ArithDialect *dialect) {
84+
SelectOp::attachInterface<SelectOpInterface>(*ctx);
85+
});
8586
}

‎mlir/lib/Dialect/Arith/Transforms/BufferViewFlowOpInterfaceImpl.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@ struct SelectOpInterface
3838

3939
void arith::registerBufferViewFlowOpInterfaceExternalModels(
4040
DialectRegistry &registry) {
41-
registry.addExtension(+[](MLIRContext *ctx, arith::ArithDialect *dialect) {
42-
SelectOp::attachInterface<SelectOpInterface>(*ctx);
43-
});
41+
registry.addExtension(
42+
"ARITH_BUFFER_FLOW", +[](MLIRContext *ctx, arith::ArithDialect *dialect) {
43+
SelectOp::attachInterface<SelectOpInterface>(*ctx);
44+
});
4445
}

‎mlir/lib/Dialect/Arith/Transforms/BufferizableOpInterfaceImpl.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -206,9 +206,10 @@ struct SelectOpInterface
206206

207207
void mlir::arith::registerBufferizableOpInterfaceExternalModels(
208208
DialectRegistry &registry) {
209-
registry.addExtension(+[](MLIRContext *ctx, ArithDialect *dialect) {
210-
ConstantOp::attachInterface<ConstantOpInterface>(*ctx);
211-
IndexCastOp::attachInterface<IndexCastOpInterface>(*ctx);
212-
SelectOp::attachInterface<SelectOpInterface>(*ctx);
213-
});
209+
registry.addExtension(
210+
"ARITH_BUFFERIZATION", +[](MLIRContext *ctx, ArithDialect *dialect) {
211+
ConstantOp::attachInterface<ConstantOpInterface>(*ctx);
212+
IndexCastOp::attachInterface<IndexCastOpInterface>(*ctx);
213+
SelectOp::attachInterface<SelectOpInterface>(*ctx);
214+
});
214215
}

‎mlir/lib/Dialect/Bufferization/TransformOps/BufferizationTransformOps.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,8 @@ class BufferizationTransformDialectExtension
161161
#include "mlir/Dialect/Bufferization/TransformOps/BufferizationTransformOps.cpp.inc"
162162
>();
163163
}
164+
165+
static constexpr llvm::StringRef extensionID = "BUFFERIZATION_TRANSFORM";
164166
};
165167
} // namespace
166168

‎mlir/lib/Dialect/Bufferization/Transforms/FuncBufferizableOpInterfaceImpl.cpp

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ getBufferizedFunctionArgType(FuncOp funcOp, int64_t index,
8282

8383
/// Return the FuncOp called by `callOp`.
8484
static FuncOp getCalledFunction(CallOpInterface callOp) {
85-
SymbolRefAttr sym = llvm::dyn_cast_if_present<SymbolRefAttr>(callOp.getCallableForCallee());
85+
SymbolRefAttr sym =
86+
llvm::dyn_cast_if_present<SymbolRefAttr>(callOp.getCallableForCallee());
8687
if (!sym)
8788
return nullptr;
8889
return dyn_cast_or_null<FuncOp>(
@@ -489,9 +490,10 @@ struct FuncOpInterface
489490

490491
void mlir::bufferization::func_ext::
491492
registerBufferizableOpInterfaceExternalModels(DialectRegistry &registry) {
492-
registry.addExtension(+[](MLIRContext *ctx, func::FuncDialect *dialect) {
493-
func::CallOp::attachInterface<func_ext::CallOpInterface>(*ctx);
494-
func::FuncOp::attachInterface<func_ext::FuncOpInterface>(*ctx);
495-
func::ReturnOp::attachInterface<func_ext::ReturnOpInterface>(*ctx);
496-
});
493+
registry.addExtension(
494+
"FUNC_BUFFERIZATION", +[](MLIRContext *ctx, func::FuncDialect *dialect) {
495+
func::CallOp::attachInterface<func_ext::CallOpInterface>(*ctx);
496+
func::FuncOp::attachInterface<func_ext::FuncOpInterface>(*ctx);
497+
func::ReturnOp::attachInterface<func_ext::ReturnOpInterface>(*ctx);
498+
});
497499
}

‎mlir/lib/Dialect/ControlFlow/Transforms/BufferDeallocationOpInterfaceImpl.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,8 @@ struct CondBranchOpInterface
157157

158158
void mlir::cf::registerBufferDeallocationOpInterfaceExternalModels(
159159
DialectRegistry &registry) {
160-
registry.addExtension(+[](MLIRContext *ctx, ControlFlowDialect *dialect) {
161-
CondBranchOp::attachInterface<CondBranchOpInterface>(*ctx);
162-
});
160+
registry.addExtension(
161+
"CF_BUFFER_DEALLOC", +[](MLIRContext *ctx, ControlFlowDialect *dialect) {
162+
CondBranchOp::attachInterface<CondBranchOpInterface>(*ctx);
163+
});
163164
}

‎mlir/lib/Dialect/ControlFlow/Transforms/BufferizableOpInterfaceImpl.cpp

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,10 @@ struct CondBranchOpInterface
6363

6464
void mlir::cf::registerBufferizableOpInterfaceExternalModels(
6565
DialectRegistry &registry) {
66-
registry.addExtension(+[](MLIRContext *ctx, cf::ControlFlowDialect *dialect) {
67-
cf::BranchOp::attachInterface<BranchOpInterface>(*ctx);
68-
cf::CondBranchOp::attachInterface<CondBranchOpInterface>(*ctx);
69-
});
66+
registry.addExtension(
67+
"CF_BUFFERIZATION",
68+
+[](MLIRContext *ctx, cf::ControlFlowDialect *dialect) {
69+
cf::BranchOp::attachInterface<BranchOpInterface>(*ctx);
70+
cf::CondBranchOp::attachInterface<CondBranchOpInterface>(*ctx);
71+
});
7072
}

‎mlir/lib/Dialect/Func/Extensions/InlinerExtension.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,11 @@ struct FuncInlinerInterface : public DialectInlinerInterface {
8080
//===----------------------------------------------------------------------===//
8181

8282
void mlir::func::registerInlinerExtension(DialectRegistry &registry) {
83-
registry.addExtension(+[](MLIRContext *ctx, func::FuncDialect *dialect) {
84-
dialect->addInterfaces<FuncInlinerInterface>();
83+
registry.addExtension(
84+
"FUNC_INLINER", +[](MLIRContext *ctx, func::FuncDialect *dialect) {
85+
dialect->addInterfaces<FuncInlinerInterface>();
8586

86-
// The inliner extension relies on the ControlFlow dialect.
87-
ctx->getOrLoadDialect<cf::ControlFlowDialect>();
88-
});
87+
// The inliner extension relies on the ControlFlow dialect.
88+
ctx->getOrLoadDialect<cf::ControlFlowDialect>();
89+
});
8990
}

‎mlir/lib/Dialect/Func/Extensions/MeshShardingExtensions.cpp

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,12 @@
1414
namespace mlir::func {
1515

1616
void registerShardingInterfaceExternalModels(DialectRegistry &registry) {
17-
registry.addExtension(+[](MLIRContext *ctx, FuncDialect *dialect) {
18-
ReturnOp::attachInterface<
19-
mesh::IndependentParallelIteratorDomainShardingInterface<ReturnOp>>(
20-
*ctx);
21-
});
17+
registry.addExtension(
18+
"FUNC_SHARDING", +[](MLIRContext *ctx, FuncDialect *dialect) {
19+
ReturnOp::attachInterface<
20+
mesh::IndependentParallelIteratorDomainShardingInterface<ReturnOp>>(
21+
*ctx);
22+
});
2223
}
2324

2425
} // namespace mlir::func

‎mlir/lib/Dialect/Func/TransformOps/FuncTransformOps.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,8 @@ class FuncTransformDialectExtension
246246
#include "mlir/Dialect/Func/TransformOps/FuncTransformOps.cpp.inc"
247247
>();
248248
}
249+
250+
static constexpr llvm::StringRef extensionID = "FUNC_TRANSFORM";
249251
};
250252
} // namespace
251253

‎mlir/lib/Dialect/GPU/TransformOps/GPUTransformOps.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -933,6 +933,8 @@ class GPUTransformDialectExtension
933933
#include "mlir/Dialect/GPU/TransformOps/GPUTransformOps.cpp.inc"
934934
>();
935935
}
936+
937+
static constexpr llvm::StringRef extensionID = "GPU_TRANSFORM";
936938
};
937939
} // namespace
938940

‎mlir/lib/Dialect/GPU/Transforms/BufferDeallocationOpInterfaceImpl.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,8 @@ struct GPUTerminatorOpInterface
3131

3232
void mlir::gpu::registerBufferDeallocationOpInterfaceExternalModels(
3333
DialectRegistry &registry) {
34-
registry.addExtension(+[](MLIRContext *ctx, GPUDialect *dialect) {
35-
gpu::TerminatorOp::attachInterface<GPUTerminatorOpInterface>(*ctx);
36-
});
34+
registry.addExtension(
35+
"GPU_BUFFER_DEALLOC", +[](MLIRContext *ctx, GPUDialect *dialect) {
36+
gpu::TerminatorOp::attachInterface<GPUTerminatorOpInterface>(*ctx);
37+
});
3738
}

0 commit comments

Comments
 (0)