init commit for sql dialect

carlguo866 · carlguo866 · commit 0572db905f53 · 2023-01-31T17:52:44.000-05:00
diff --git a/include/CMakeLists.txt b/include/CMakeLists.txt
@@ -1 +1,2 @@
 add_subdirectory(polygeist)
+add_subdirectory(sql)
diff --git a/include/sql/CMakeLists.txt b/include/sql/CMakeLists.txt
@@ -0,0 +1,3 @@
+add_mlir_dialect(SQLOps sql)
+# add_mlir_doc(SQLDialect -gen-dialect-doc SQLDialect SQL/)
+# add_mlir_doc(SQLOps -gen-op-doc SQLOps SQL/)
diff --git a/include/sql/SQLDialect.h b/include/sql/SQLDialect.h
@@ -0,0 +1,16 @@
+//===- BFVDialect.h - BFV dialect -----------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SQL_DIALECT
+#define SQL_DIALECT
+
+#include "mlir/IR/Dialect.h"
+
+#include "sql/SQLOpsDialect.h.inc"
+
+#endif 
diff --git a/include/sql/SQLDialect.td b/include/sql/SQLDialect.td
@@ -0,0 +1,44 @@
+//===- BFVDialect.td - BFV dialect -----------*- tablegen -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SQL_DIALECT
+#define SQL_DIALECT
+
+
+include "mlir/IR/OpBase.td"
+include "mlir/IR/FunctionInterfaces.td"
+include "mlir/IR/SymbolInterfaces.td"
+include "mlir/Interfaces/SideEffectInterfaces.td"
+
+include "mlir/Interfaces/CallInterfaces.td"
+include "mlir/Interfaces/CastInterfaces.td"
+
+
+def SQL_Dialect : Dialect {
+  let summary = "A dialect for SQL languages in MLIR.";
+  let description = [{
+    TBD
+  }];
+  let name = "sql";
+  let cppNamespace = "::mlir::sql";
+}
+
+
+//===----------------------------------------------------------------------===//
+// SQL Operations
+//===----------------------------------------------------------------------===//
+
+class SQL_Op<string mnemonic, list<Trait> traits = []>
+    : Op<SQL_Dialect, mnemonic, traits>; 
+
+#endif // SQL_DIALECT 
+
+
+
+
+
diff --git a/include/sql/SQLOps.h b/include/sql/SQLOps.h
@@ -0,0 +1,23 @@
+//===- Polygeistps.h - Polygeist dialect ops --------------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SQLOPS_H
+#define SQLOPS_H
+
+#include "mlir/IR/BuiltinTypes.h"
+#include "mlir/IR/Dialect.h"
+#include "mlir/IR/OpDefinition.h"
+#include "mlir/IR/PatternMatch.h"
+#include "mlir/Interfaces/SideEffectInterfaces.h"
+#include "mlir/Interfaces/ViewLikeInterface.h"
+#include "llvm/Support/CommandLine.h"
+
+#define GET_OP_CLASSES
+#include "sql/SQLOps.h.inc"
+
+#endif
diff --git a/include/sql/SQLOps.td b/include/sql/SQLOps.td
@@ -0,0 +1,57 @@
+//===- SQLOps.td - Polygeist dialect ops ----------------*- tablegen -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+//  #ifndef SQL_OPS
+// #define SQL_OPS
+
+include "mlir/IR/AttrTypeBase.td"
+include "SQLDialect.td"
+
+def SelectOp : SQL_Op<"select", [Pure]> {
+  let summary = "select";
+
+  // TODO: limit (optional), where clauses, join, etc
+  let arguments = (ins StrArrayAttr:$column, StrAttr:$table);
+  let results = (outs Index : $result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+} 
+
+def ExecuteOp : SQL_Op<"execute", []> {
+  let summary = "execute query";
+
+  let arguments = (ins Index:$handle);
+  let results = (outs Index:$result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+}
+
+def NumResultsOp : SQL_Op<"num_results", [Pure]> {
+  let summary = "number of results";
+
+  let arguments = (ins Index:$handle);
+  let results = (outs Index:$result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+}  
+
+def ResultOp : SQL_Op<"get_result", [Pure]> {
+  let summary = "get results of execution";
+
+  let arguments = (ins Index:$handle, StrAttr:$column, Index:$row);
+  let results = (outs AnyType:$result);
+
+  let hasFolder = 0;
+  let hasCanonicalizer = 0;
+}  
+ 
+
+// #endif
diff --git a/lib/CMakeLists.txt b/lib/CMakeLists.txt
@@ -1 +1,2 @@
 add_subdirectory(polygeist)
+add_subdirectory(sql)
diff --git a/lib/sql/CMakeLists.txt b/lib/sql/CMakeLists.txt
@@ -0,0 +1,19 @@
+add_mlir_dialect_library(MLIRSQL
+Dialect.cpp
+Ops.cpp
+
+ADDITIONAL_HEADER_DIRS
+${PROJECT_SOURCE_DIR}/include/sql
+
+DEPENDS
+MLIRSQLOpsIncGen
+
+LINK_LIBS PUBLIC
+MLIRIR
+MLIRMemRefDialect
+MLIRLLVMDialect
+MLIROpenMPDialect
+MLIRAffineDialect
+MLIRSupport
+MLIRSCFTransforms
+)
diff --git a/lib/sql/Dialect.cpp b/lib/sql/Dialect.cpp
@@ -0,0 +1,27 @@
+//===- PolygeistDialect.cpp - Polygeist dialect ---------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "sql/SQLDialect.h"
+#include "mlir/IR/DialectImplementation.h"
+#include "sql/SQLOps.h"
+
+using namespace mlir;
+using namespace mlir::sql;
+
+//===----------------------------------------------------------------------===//
+// Polygeist dialect.
+//===----------------------------------------------------------------------===//
+
+void SQLDialect::initialize() {
+  addOperations<
+#define GET_OP_LIST
+#include "sql/SQLOps.cpp.inc"
+      >();
+}
+
+#include "sql/SQLOpsDialect.cpp.inc"
diff --git a/lib/sql/Ops.cpp b/lib/sql/Ops.cpp
@@ -0,0 +1,42 @@
+//===- PolygeistOps.cpp - BFV dialect ops ---------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "sql/SQLOps.h"
+#include "mlir/Dialect/Arith/IR/Arith.h"
+#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
+#include "mlir/Dialect/LLVMIR/LLVMTypes.h"
+#include "mlir/IR/AffineExpr.h"
+#include "mlir/IR/Builders.h"
+#include "mlir/IR/OpImplementation.h"
+#include "mlir/Interfaces/SideEffectInterfaces.h"
+#include "sql/SQLDialect.h"
+
+#define GET_OP_CLASSES
+#include "sql/SQLOps.cpp.inc"
+
+#include "mlir/Dialect/Affine/IR/AffineOps.h"
+#include "mlir/Dialect/Arith/Utils/Utils.h"
+#include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/Dialect/MemRef/IR/MemRef.h"
+#include "mlir/Dialect/OpenMP/OpenMPDialect.h"
+#include "mlir/Dialect/SCF/IR/SCF.h"
+#include "mlir/IR/BlockAndValueMapping.h"
+#include "mlir/IR/Dominance.h"
+#include "mlir/IR/IntegerSet.h"
+#include "mlir/Transforms/SideEffectUtils.h"
+
+#include "llvm/ADT/SetVector.h"
+#include "llvm/Support/Debug.h"
+
+#define DEBUG_TYPE "sql"
+
+using namespace mlir;
+using namespace sql;
+using namespace mlir::arith;
+
+
diff --git a/test/polygeist-opt/sql.mlir b/test/polygeist-opt/sql.mlir
@@ -0,0 +1,11 @@
+// RUN: polygeist-opt %s | FileCheck %s
+
+module  {
+  func.func private @run() -> i32 {
+    %c0 = arith.constant 0 : index
+    %q = "sql.select"() {column = ["data"], table = "mytable"} : () -> index
+    %h = "sql.execute"(%q) : (index) -> index
+    %res = "sql.get_result"(%h, %c0) {column = "data"} : (index, index) -> i32
+    return %res : i32
+  }
+}
diff --git a/tools/polygeist-opt/CMakeLists.txt b/tools/polygeist-opt/CMakeLists.txt
@@ -6,6 +6,7 @@ set(LIBS
         MLIROptLib
 		MLIRPolygeist
 		MLIRPolygeistTransforms
+        MLIRSQL
         )
 add_llvm_executable(polygeist-opt polygeist-opt.cpp)
 
diff --git a/tools/polygeist-opt/polygeist-opt.cpp b/tools/polygeist-opt/polygeist-opt.cpp
@@ -32,6 +32,9 @@
 #include "polygeist/Dialect.h"
 #include "polygeist/Passes/Passes.h"
 
+#include "sql/SQLDialect.h"
+#include "sql/SQLOps.h"
+
 using namespace mlir;
 
 class MemRefInsider
@@ -60,6 +63,7 @@ int main(int argc, char **argv) {
   registry.insert<DLTIDialect>();
 
   registry.insert<mlir::polygeist::PolygeistDialect>();
+  registry.insert<mlir::sql::SQLDialect>();
 
   mlir::registerpolygeistPasses();
 
@@ -74,6 +78,7 @@ int main(int argc, char **argv) {
   mlir::registerConvertSCFToOpenMPPass();
   mlir::registerAffinePasses();
 
+
   registry.addExtension(+[](MLIRContext *ctx, LLVM::LLVMDialect *dialect) {
     LLVM::LLVMFunctionType::attachInterface<MemRefInsider>(*ctx);
   });

Original file line number	Diff line number	Diff line change
`@@ -1 +1,2 @@`
`1`	`1`	`add_subdirectory(polygeist)`
	`2`	`+add_subdirectory(sql)`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+add_mlir_dialect(SQLOps sql)`
	`2`	`+# add_mlir_doc(SQLDialect -gen-dialect-doc SQLDialect SQL/)`
	`3`	`+# add_mlir_doc(SQLOps -gen-op-doc SQLOps SQL/)`