Provide predicated data type

include/Conversion/ConversionPasses.h

@@ -12,11 +12,11 @@
 
 namespace mlir {
 
-// Passes defined in GraphPasses.td
+// Passes defined in GraphPasses.td.
 #define GEN_PASS_DECL
 #include "Conversion/ConversionPasses.h.inc"
 
-// Conversion passes
+// Conversion passes.
 std::unique_ptr<mlir::Pass> createLowerArithToNeuraPass();
 std::unique_ptr<mlir::Pass> createLowerLlvmToNeuraPass();
 

include/NeuraDialect/CMakeLists.txt

@@ -1,3 +1,11 @@
+# Set TableGen include paths
+set(MLIR_TABLEGEN_INCLUDES 
+    ${PROJECT_SOURCE_DIR}/include
+    ${PROJECT_SOURCE_DIR}/include/NeuraDialect
+    ${CMAKE_CURRENT_BINARY_DIR}/include/NeuraDialect
+    ${MLIR_MAIN_INCLUDE_DIR}
+    ${MLIR_INCLUDE_DIR})
+
 add_mlir_dialect(Neura neura)
 
 set(LLVM_TARGET_DEFINITIONS NeuraPasses.td)

include/NeuraDialect/Neura.td

@@ -4,5 +4,6 @@
 include "NeuraDialect.td"
 include "NeuraOps.td"
 include "NeuraPasses.td"
+include "NeuraTypes.td"
 
 #endif // GRAPH_TD

include/NeuraDialect/NeuraDialect.h

@@ -1,17 +1,34 @@
-#ifndef NEURADIALECT_NEURADIALECT_H
-#define NEURADIALECT_NEURADIALECT_H
+#ifndef NEURA_DIALECT_H
+#define NEURA_DIALECT_H
 
 #include "mlir/IR/Dialect.h"
 #include "mlir/IR/OpDefinition.h"
 
-// Defines the export macro.
 #ifdef _WIN32
-  #define NEURA_DIALECT_EXPORT __declspec(dllexport)
+#define NEURA_DIALECT_EXPORT __declspec(dllexport)
 #else
-  #define NEURA_DIALECT_EXPORT __attribute__((visibility("default")))
+#define NEURA_DIALECT_EXPORT __attribute__((visibility("default")))
 #endif
 
-// Includes generated TableGen headers.
+namespace mlir {
+namespace neura {
+
+// Forward declare before including generated code
+class NeuraDialect;
+
+} // end namespace neura
+} // end namespace mlir
+
+// Include the generated dialect declarations
 #include "NeuraDialect/NeuraDialect.h.inc"
 
-#endif // NEURADIALECT_NEURADIALECT_H
+namespace mlir {
+namespace neura {
+
+// Declare additional methods for the generated dialect class
+NEURA_DIALECT_EXPORT void registerNeuraDialect();
+
+} // end namespace neura
+} // end namespace mlir
+
+#endif // NEURA_DIALECT_H

include/NeuraDialect/NeuraDialect.td

@@ -1,13 +1,33 @@
 // NeuraDialect.td - TableGen description of the dialect.
-#ifndef NEURADIALECT_TD
-#define NEURADIALECT_TD
+#ifndef NEURA_DIALECT_TD
+#define NEURA_DIALECT_TD
 
 include "mlir/IR/OpBase.td"
 include "mlir/IR/DialectBase.td"
+include "mlir/IR/AttrTypeBase.td"
 
+// First define the dialect
 def NeuraDialect : Dialect {
   let name = "neura";
   let cppNamespace = "::mlir::neura";
+
+  let summary = "A dialect for the Neura compiler infrastructure.";
+  let description = [{
+    This dialect is used for representing Neura operations and types.
+  }];
+
+  let useDefaultTypePrinterParser = 0;
+  let useDefaultAttributePrinterParser = 0;
+
+  let extraClassDeclaration = [{
+  // Type parsing/printing
+  Type parseType(DialectAsmParser &parser) const override;
+  void printType(Type type, DialectAsmPrinter &printer) const override;
+
+  // Attribute parsing/printing
+  Attribute parseAttribute(DialectAsmParser &parser, Type type) const override;
+  void printAttribute(Attribute attr, DialectAsmPrinter &printer) const override;
+}];
 }
 
-#endif // NEURADIALECT_TD
+#endif // NEURA_DIALECT_TD

include/NeuraDialect/NeuraOps.h

@@ -7,9 +7,19 @@
 #include "mlir/IR/DialectImplementation.h"
 #include "mlir/IR/Builders.h"
 
-#define GET_OP_CLASSES
+// First includes the interface declarations.
+#define GET_OP_INTERFACE_CLASSES
+#include "NeuraDialect/Neura.h.inc"
+#undef GET_OP_INTERFACE_CLASSES
+
+// Then includes the op declarations.
+#define GET_OP_DECLARATIONS
 #include "NeuraDialect/Neura.h.inc"
+#undef GET_OP_DECLARATIONS
 
-// Additional definitions or includes can go here.
+// Finally includes the op definitions.
+#define GET_OP_CLASSES
+#include "NeuraDialect/Neura.h.inc"
+#undef GET_OP_CLASSES
 
 #endif // NEURA_OPS_H