[CIR][MLIR] named_tuple.cast operation

WIP memref cast operation for NamedTuple hack.
The `named_tuple.cast` operation converts a memref of `named_tuple` to a 1D
memref of `i8` of the same size to emulate later the struct element access.

    Example:
    %0 = named_tuple.cast %alloca_0 : memref<!named_tuple.named_tuple<"s", [i32, i32]>> to memref<8xi8> loc(#loc6)

Author: keryell

mlir/include/mlir/Dialect/NamedTuple/IR/NamedTuple.h

@@ -13,9 +13,11 @@
 #ifndef MLIR_DIALECT_NAMED_TUPLE_IR_NAMED_TUPLE_H
 #define MLIR_DIALECT_NAMED_TUPLE_IR_NAMED_TUPLE_H
 
+//#include "mlir/IR/Dialect.h"
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleDialect.h"
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleTypes.h"
 
+#define GET_OP_CLASSES
 #include "mlir/Dialect/NamedTuple/IR/NamedTuple.h.inc"
 
 #endif // MLIR_DIALECT_NAMED_TUPLE_IR_NAMED_TUPLE_H

mlir/include/mlir/Dialect/NamedTuple/IR/NamedTupleOps.td

@@ -9,6 +9,37 @@
 #ifndef NAMED_TUPLE_OPS
 #define NAMED_TUPLE_OPS
 
-include "mlir/IR/OpAsmInterface.td"
+include "mlir/Dialect/NamedTuple/IR/NamedTupleDialect.td"
+include "mlir/Dialect/Arith/IR/ArithBase.td"
+include "mlir/Interfaces/CastInterfaces.td"
+include "mlir/Interfaces/SideEffectInterfaces.td"
+include "mlir/IR/OpBase.td"
+
+class NamedTuple_Op<string mnemonic, list<Trait> traits = []>
+    : Op<NamedTuple_Dialect, mnemonic, traits>;
+
+//===----------------------------------------------------------------------===//
+// CastOp
+//===----------------------------------------------------------------------===//
+
+def NamedTuple_CastOp : NamedTuple_Op<"cast", [
+      DeclareOpInterfaceMethods<CastOpInterface>,
+      Pure,
+    ]> {
+  let summary = "WIP memref cast operation for NamedTuple hack";
+  let description = [{
+    The `named_tuple.cast` operation converts a memref of `named_tuple` to a 1D memref of `i8` of the same size to emulate later the struct element access.
+
+    Example:
+
+    ```mlir
+    %0 = named_tuple.cast %alloca_0 : memref<!named_tuple.named_tuple<"s", [i32, i32]>> to memref<8xi8> loc(#loc6)
+    ```
+  }];
+
+  let arguments = (ins AnyRankedOrUnrankedMemRef:$source);
+  let results = (outs AnyRankedOrUnrankedMemRef:$dest);
+  let assemblyFormat = "$source attr-dict `:` type($source) `to` type($dest)";
+}
 
 #endif // NAMED_TUPLE_OPS

mlir/include/mlir/Dialect/NamedTuple/IR/NamedTupleTypes.h

@@ -13,9 +13,8 @@
 #ifndef MLIR_DIALECT_NAMED_TUPLE_IR_NAMED_TUPLE_TYPES_H
 #define MLIR_DIALECT_NAMED_TUPLE_IR_NAMED_TUPLE_TYPES_H
 
-//#include "mlir/Bytecode/BytecodeOpInterface.h"
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleDialect.h"
-//#include "mlir/IR/OpDefinition.h"
+#include "mlir/Interfaces/DataLayoutInterfaces.h"
 
 #define GET_TYPEDEF_CLASSES
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleTypes.h.inc"

mlir/include/mlir/Dialect/NamedTuple/IR/NamedTupleTypes.td

@@ -27,12 +27,10 @@ class NamedTuple_Type<string name, string typeMnemonic, list<Trait> traits = []>
 }
 
 def NamedTuple_NamedTupleType : NamedTuple_Type<"NamedTuple", "named_tuple", [
-    MemRefElementTypeInterface//,
-    //DeclareTypeInterfaceMethods<DataLayoutTypeInterface, [
-    //  "areCompatible", "getIndexBitwidth", "verifyEntries"]>
+    MemRefElementTypeInterface,
+    DeclareTypeInterfaceMethods<DataLayoutTypeInterface>
   ]> {
-  let summary = "Fixed-sized collection of other types with a name for each"
-      " type";
+  let summary = "Fixed-sized collection of other types with a name";
   let description = [{
   }];
   let parameters = (ins "StringAttr":$name, Builtin_ArrayAttr:$member_types);
@@ -41,9 +39,9 @@ def NamedTuple_NamedTupleType : NamedTuple_Type<"NamedTuple", "named_tuple", [
     TypeBuilder<(ins "StringRef":$name, "ArrayRef<Type>":$types), [{
       llvm::SmallVector<mlir::Attribute> typeAttrs;
       for (auto &t : types)
-      typeAttrs.push_back(mlir::TypeAttr::get(t));
-       return $_get($_ctxt, mlir::StringAttr::get($_ctxt, name),
-                    mlir::ArrayAttr::get($_ctxt, typeAttrs));
+        typeAttrs.push_back(mlir::TypeAttr::get(t));
+      return $_get($_ctxt, mlir::StringAttr::get($_ctxt, name),
+                   mlir::ArrayAttr::get($_ctxt, typeAttrs));
     }]>,
     // Empty tuple
     TypeBuilder<(ins "StringRef":$name), [{
@@ -53,6 +51,7 @@ def NamedTuple_NamedTupleType : NamedTuple_Type<"NamedTuple", "named_tuple", [
     }]>
   ];
 
+  // \todo
   // let extraClassDeclaration = [{
   //   /// Return the number of held types.
   //   size_t size() const;

mlir/lib/Dialect/NamedTuple/IR/CMakeLists.txt

@@ -1,6 +1,5 @@
 add_mlir_dialect_library(MLIRNamedTupleDialect
   NamedTupleDialect.cpp
-  NamedTupleOps.cpp
 
   ADDITIONAL_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}/include/mlir/Dialect/NamedTuple

mlir/lib/Dialect/NamedTuple/IR/NamedTupleDialect.cpp

@@ -10,16 +10,52 @@
 //
 //===----------------------------------------------------------------------===//
 
-#include "llvm/ADT/TypeSwitch.h"
+#include "mlir/Dialect/MemRef/IR/MemRef.h"
 #include "mlir/Dialect/NamedTuple/IR/NamedTuple.h"
 #include "mlir/IR/Builders.h"
 #include "mlir/IR/DialectImplementation.h"
+#include "llvm/ADT/TypeSwitch.h"
+#include "llvm/Support/ErrorHandling.h"
 
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleDialect.cpp.inc"
 
 #define GET_TYPEDEF_CLASSES
 #include "mlir/Dialect/NamedTuple/IR/NamedTupleTypes.cpp.inc"
 
+// Dummy implementation for now
+// TODO(cir)
+llvm::TypeSize mlir::named_tuple::NamedTupleType::getTypeSizeInBits(
+    mlir::DataLayout const &,
+    llvm::ArrayRef<mlir::DataLayoutEntryInterface>) const {
+  llvm_unreachable("getTypeSizeInBits() not implemented");
+  return llvm::TypeSize::getFixed(8);
+}
+
+uint64_t mlir::named_tuple::NamedTupleType::getABIAlignment(
+    mlir::DataLayout const &,
+    llvm::ArrayRef<mlir::DataLayoutEntryInterface>) const {
+  llvm_unreachable("getABIAlignment() not implemented");
+  return 8;
+}
+
+uint64_t mlir::named_tuple::NamedTupleType::getPreferredAlignment(
+    mlir::DataLayout const &,
+    llvm::ArrayRef<mlir::DataLayoutEntryInterface>) const {
+  llvm_unreachable("getPreferredAlignment() not implemented");
+  return 8;
+}
+
+//===----------------------------------------------------------------------===//
+// TableGen'd op method definitions
+//===----------------------------------------------------------------------===//
+#define GET_OP_CLASSES
+#include "mlir/Dialect/NamedTuple/IR/NamedTuple.cpp.inc"
+
+bool mlir::named_tuple::CastOp::areCastCompatible(mlir::TypeRange inputs,
+                                                  mlir::TypeRange outputs) {
+  return true;
+}
+
 void mlir::named_tuple::NamedTupleDialect::initialize() {
   // Add the defined types to the dialect.
   addTypes<