[MLIR] Initial upstream of polygeist dialect

Author: wsmoses

.github/CODEOWNERS

@@ -122,6 +122,9 @@
 /mlir/lib/Dialect/Vector/Transforms/VectorEmulateNarrowType.cpp @banach-space @dcaballe @MaheshRavishankar @nicolasvasilache
 /mlir/**/*EmulateNarrowType* @dcaballe
 
+# Polygeist dialect in MLIR
+/mlir/**/*Polygeist* @wsmoses @ftynse @ivanradanov @chelini
+
 # Presburger library in MLIR
 /mlir/**/*Presburger* @Groverkss @Superty
 

mlir/include/mlir/Dialect/CMakeLists.txt

@@ -27,6 +27,7 @@ add_subdirectory(OpenACCMPCommon)
 add_subdirectory(OpenMP)
 add_subdirectory(PDL)
 add_subdirectory(PDLInterp)
+add_subdirectory(Polygeist)
 add_subdirectory(Ptr)
 add_subdirectory(Quant)
 add_subdirectory(SCF)

mlir/include/mlir/Dialect/Polygeist/CMakeLists.txt

@@ -0,0 +1 @@
+add_subdirectory(IR)

mlir/include/mlir/Dialect/Polygeist/IR/CMakeLists.txt

@@ -0,0 +1,3 @@
+set(LLVM_TARGET_DEFINITIONS PolygeistOps.td)
+add_mlir_dialect(PolygeistOps polygeist)
+add_mlir_doc(PolygeistOps PolygeistOps Dialects/ -gen-dialect-doc)

mlir/include/mlir/Dialect/Polygeist/IR/Polygeist.h

@@ -0,0 +1,15 @@
+#ifndef MLIR_DIALECT_POLYGEIST_IR_POLYGEIST_H_
+#define MLIR_DIALECT_POLYGEIST_IR_POLYGEIST_H_
+
+#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
+#include "mlir/IR/Dialect.h"
+#include "mlir/IR/OpDefinition.h"
+#include "mlir/Interfaces/SideEffectInterfaces.h"
+#include "mlir/Interfaces/ViewLikeInterface.h"
+
+#include "mlir/Dialect/Polygeist/IR/PolygeistOpsDialect.h.inc"
+
+#define GET_OP_CLASSES
+#include "mlir/Dialect/Polygeist/IR/PolygeistOps.h.inc"
+
+#endif // MLIR_DIALECT_POLYGEIST_IR_POLYGEIST_H_

mlir/include/mlir/Dialect/Polygeist/IR/PolygeistBase.td

@@ -0,0 +1,15 @@
+#ifndef POLYGEIST_BASE
+#define POLYGEIST_BASE
+
+include "mlir/IR/OpBase.td"
+
+def Polygeist_Dialect : Dialect {
+  let name = "polygeist";
+  let cppNamespace = "::mlir::polygeist";
+  let summary = "The Polygeist dialect.";
+  let description = [{
+    The Polygeist dialect contains operations for raising low-level code to higher-level forms, and performing parallel and device transformations (including polyhedral).
+  }];
+}
+
+#endif // POLYGEIST_BASE

mlir/include/mlir/Dialect/Polygeist/IR/PolygeistOps.td

@@ -0,0 +1,40 @@
+#ifndef POLYGEIST_OPS
+#define POLYGEIST_OPS
+
+include "mlir/IR/OpBase.td"
+include "mlir/Interfaces/ViewLikeInterface.td"
+include "mlir/Interfaces/SideEffectInterfaces.td"
+include "mlir/Dialect/LLVMIR/LLVMOpBase.td"
+include "mlir/Dialect/Polygeist/IR/PolygeistBase.td"
+
+def Memref2PointerOp
+    : Op<Polygeist_Dialect, "memref2pointer", [ViewLikeOpInterface, Pure]> {
+  let summary = "Extract an LLVM pointer from a MemRef";
+
+  let arguments = (ins AnyMemRef:$source);
+  let results = (outs LLVM_AnyPointer:$result);
+
+  let hasFolder = 1;
+  let hasCanonicalizer = 1;
+
+  let extraClassDeclaration = [{
+    ::mlir::Value getViewSource() { return getSource(); }
+  }];
+}
+
+def Pointer2MemrefOp
+    : Op<Polygeist_Dialect, "pointer2memref", [ViewLikeOpInterface, Pure]> {
+  let summary = "Upgrade a pointer to a memref";
+
+  let arguments = (ins LLVM_AnyPointer:$source);
+  let results = (outs AnyMemRef:$result);
+
+  let hasFolder = 1;
+  let hasCanonicalizer = 1;
+
+  let extraClassDeclaration = [{
+    ::mlir::Value getViewSource() { return getSource(); }
+  }];
+}
+
+#endif // POLYGEIST_OPS

mlir/lib/Dialect/CMakeLists.txt

@@ -27,6 +27,7 @@ add_subdirectory(OpenACCMPCommon)
 add_subdirectory(OpenMP)
 add_subdirectory(PDL)
 add_subdirectory(PDLInterp)
+add_subdirectory(Polygeist)
 add_subdirectory(Ptr)
 add_subdirectory(Quant)
 add_subdirectory(SCF)

mlir/lib/Dialect/Polygeist/CMakeLists.txt

@@ -0,0 +1 @@
+add_subdirectory(IR)

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

@@ -0,0 +1,19 @@
+add_mlir_dialect_library(MLIRPolygeistDialect
+  PolygeistOps.cpp
+  PolygeistDialect.cpp
+
+  ADDITIONAL_HEADER_DIRS
+  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Polygeist
+
+  DEPENDS
+  MLIRPolygeistOpsIncGen
+
+  LINK_LIBS PUBLIC
+  MLIRDialect
+  MLIRIR
+  MLIRMemRefDialect
+  MLIRLLVMDialect
+  MLIRArithDialect
+  MLIRAffineDialect
+  MLIRSCFDialect
+  )