Create infrastructure for SYCLToLLVM conversion (#37)

Signed-off-by: Tiotto, Ettore <[email protected]>

Author: etiotto

mlir-sycl/CMakeLists.txt

@@ -67,7 +67,7 @@ set(MLIR_SYCL_TOOLS_DIR ${CMAKE_BINARY_DIR}/bin)
 set(LLVM_LIT_ARGS "-sv" CACHE STRING "lit default options")
 list(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake/modules")
 
-add_subdirectory(include)
+add_subdirectory(include/mlir)
 add_subdirectory(lib)
 add_subdirectory(test)
 

mlir-sycl/include/CMakeLists.txt

@@ -0,0 +1,2 @@
+add_subdirectory(Conversion)
+add_subdirectory(Dialect)

mlir-sycl/include/SYCL/SYCLToLLVM.h

@@ -0,0 +1,3 @@
+set(LLVM_TARGET_DEFINITIONS SYCLPasses.td)
+mlir_tablegen(SYCLPasses.h.inc -gen-pass-decls -name Conversion)
+add_public_tablegen_target(MLIRSYCLConversionPassIncGen)

mlir-sycl/include/mlir/CMakeLists.txt

@@ -0,0 +1,22 @@
+//===- SYCLPasses.h - Conversion Pass Construction and Registration -----------===//
+//
+// Part of the LLVM Project, 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 MLIR_CONVERSION_SYCLPASSES_H
+#define MLIR_CONVERSION_SYCLPASSES_H
+
+#include "mlir/Conversion/SYCLToLLVM/SYCLToLLVM.h"
+
+namespace mlir {
+
+/// Generate the code for registering conversion passes.
+#define GEN_PASS_REGISTRATION
+#include "mlir/Conversion/SYCLPasses.h.inc"
+
+} // namespace mlir
+
+#endif // MLIR_CONVERSION_SYCLPASSES_H

mlir-sycl/include/mlir/Conversion/CMakeLists.txt

@@ -0,0 +1,28 @@
+//===-- SYCLPasses.td - Conversion pass definition file ----*- tablegen -*-===//
+//
+// Part of the LLVM Project, 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 MLIR_CONVERSION_SYCLPASSES
+#define MLIR_CONVERSION_SYCLPASSES
+
+include "mlir/Pass/PassBase.td"
+
+//===----------------------------------------------------------------------===//
+// SYCLToLLVM
+//===----------------------------------------------------------------------===//
+
+def ConvertSYCLToLLVM : Pass<"convert-sycl-to-llvm", "ModuleOp"> {
+  let summary = "Convert SYCL dialect to LLVM dialect";
+  let description = [{
+    See docs/SYCLToLLVMDialectConversion/ for more details.
+    TODO: add docs referenced above.
+  }];
+  let constructor = "mlir::createConvertSYCLToLLVMPass()";
+  let dependentDialects = ["LLVM::LLVMDialect"];
+}
+
+#endif // MLIR_CONVERSION_SYCLPASSES

mlir-sycl/include/mlir/Conversion/SYCLPasses.h

@@ -0,0 +1,46 @@
+//===- SYCLToLLVM.h - SYCL to LLVM Patterns ---------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, 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
+//
+//===----------------------------------------------------------------------===//
+//
+// Provides patterns to convert SYCL dialect to LLVM dialect.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVM_H
+#define MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVM_H
+
+#include "mlir/Transforms/DialectConversion.h"
+
+namespace mlir {
+class LLVMTypeConverter;
+class MLIRContext;
+class ModuleOp;
+
+namespace sycl {
+template <typename SYCLOp>
+class SYCLToLLVMConversion : public OpConversionPattern<SYCLOp> {
+public:
+  SYCLToLLVMConversion(MLIRContext *context, LLVMTypeConverter &typeConverter,
+                       PatternBenefit benefit = 1)
+      : OpConversionPattern<SYCLOp>(typeConverter, context, benefit),
+        typeConverter(typeConverter) {}
+
+protected:
+  LLVMTypeConverter &typeConverter;
+};
+
+/// Populates type conversions with additional SYCL types.
+void populateSYCLToLLVMTypeConversion(LLVMTypeConverter &typeConverter);
+
+/// Populates the given list with patterns that convert from SYCL to LLVM.
+void populateSYCLToLLVMConversionPatterns(LLVMTypeConverter &typeConverter,
+                                          RewritePatternSet &patterns);
+
+} // namespace sycl
+} // namespace mlir
+
+#endif // MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVM_H

mlir-sycl/include/mlir/Conversion/SYCLPasses.td

@@ -0,0 +1,27 @@
+//===- SYCLToLLVMPass.h - SYCL to LLVM Passes -------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, 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
+//
+//===----------------------------------------------------------------------===//
+//
+// Provides passes to convert SYCL dialect to LLVM dialect.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVMPASS_H
+#define MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVMPASS_H
+
+#include <memory>
+
+namespace mlir {
+class ModuleOp;
+template <typename T> class OperationPass;
+
+/// Creates a pass to convert SYCL operations to the LLVMIR dialect.
+std::unique_ptr<OperationPass<ModuleOp>> createConvertSYCLToLLVMPass();
+
+} // namespace mlir
+
+#endif // MLIR_CONVERSION_SYCLTOLLVM_SYCLTOLLVMPASS_H

mlir-sycl/include/mlir/Conversion/SYCLToLLVM/SYCLToLLVM.h

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