Lines Matching defs:TestDialect

18 struct TestDialect : public Dialect {
19   MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestDialect)
22   TestDialect(MLIRContext *context)
23       : Dialect(getDialectNamespace(), context, TypeID::get<TestDialect>()) {}
39   context.loadDialect<TestDialect>();
76   registry.insert<TestDialect, SecondTestDialect>();
78   // Delayed registration of an interface for TestDialect.
79   registry.addExtension(+[](MLIRContext *ctx, TestDialect *dialect) {
85   // Load the TestDialect and check that the interface got registered for it.
86   Dialect *testDialect = context.getOrLoadDialect<TestDialect>();
116   registry.insert<TestDialect>();
117   registry.addExtension(+[](MLIRContext *ctx, TestDialect *dialect) {
122   // Load the TestDialect and check that the interface got registered for it.
123   Dialect *testDialect = context.getOrLoadDialect<TestDialect>();
131   secondRegistry.insert<TestDialect>();
132   secondRegistry.addExtension(+[](MLIRContext *ctx, TestDialect *dialect) {
143 struct DummyExtension : DialectExtension<DummyExtension, TestDialect> {
149   void apply(MLIRContext *ctx, TestDialect *dialect) const final {
173   registry.insert<TestDialect>();
186   Dialect *testDialect = context.getOrLoadDialect<TestDialect>();
197   registry1.insert<TestDialect>();
198   registry2.insert<TestDialect>();
223   context.getOrLoadDialect<TestDialect>();