build driver

Author: hhkit

mlir/lib/Target/IRDLToCpp/IRDLToCpp.cpp

@@ -279,11 +279,6 @@ static LogicalResult generateLib(irdl::DialectOp dialect, raw_ostream &output,
   output << "#ifdef " << definitionMacroFlag << "\n#undef "
          << definitionMacroFlag << "\n";
 
-  output << llvm::formatv(dialectDefTemplateText, dialectStrings.namespaceOpen,
-                          dialectStrings.namespaceClose,
-                          dialectStrings.dialectCppName,
-                          dialectStrings.namespacePath);
-
   // type header
   output << llvm::formatv(
       typeHeaderDefTemplateText, dialectStrings.dialectBaseTypeName,
@@ -432,6 +427,14 @@ static void {0}::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &od
   output << llvm::formatv(opDefTemplateText, commaSeparatedOpList,
                           perOpDefinitions);
 
+  output << llvm::formatv(dialectDefTemplateText, dialectStrings.namespaceOpen,
+                          dialectStrings.namespaceClose,
+                          dialectStrings.dialectCppName,
+                          dialectStrings.namespacePath,
+                          commaSeparatedOpList,
+                          commaSeparatedTypeList
+                          );
+
   output << "#endif // " << definitionMacroFlag << "\n";
   return success();
 }

mlir/lib/Target/IRDLToCpp/Templates/DialectDef.txt

@@ -1,3 +1,14 @@
+/*
+{0}: namespace open
+{1}: namespace close
+
+{2}: Dialect Cpp Name
+{3}: Namespace path
+
+{4}: Op list
+{5}: Type list
+*/
+
 R"(
 MLIR_DEFINE_EXPLICIT_TYPE_ID({3}::{2})
 {0}
@@ -8,5 +19,15 @@ MLIR_DEFINE_EXPLICIT_TYPE_ID({3}::{2})
 }
 
 {2}::~{2}() = default;
+
+
+void {2}::initialize() {{
+    addOperations<
+      {4}
+    >();
+    addTypes<
+      {5}
+    >();
+}
 {1}
 )"

mlir/lib/Target/IRDLToCpp/Templates/Driver.txt

@@ -0,0 +1,15 @@
+/*
+{0}: Dialect CppName
+
+{1}: Namespace path
+
+{2}: Includes
+*/
+
+R"(
+
+
+
+
+
+)"

mlir/test/lib/Dialect/TestIRDLToCpp/TestIRDLToCppDialect.cpp

@@ -0,0 +1,3 @@
+
+#include "test_irdl_to_cpp.irdl.mlir.cpp.inc"
+

mlir/test/lib/Dialect/TestIRDLToCpp/test_irdl_to_cpp.irdl.mlir

@@ -1 +1,8 @@
-irdl.dialect @test_irdl_to_cpp {}
+irdl.dialect @test_irdl_to_cpp {
+    irdl.type @foo
+
+    irdl.operation @bar {
+        %0 = irdl.any
+        irdl.results(res: %0)
+    }
+}