[SYCL-MLIR]: Add convert-spirv-builtin-to-nvvm Pass

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

@@ -0,0 +1,28 @@
+//===- SPIRVBuiltInToNVVM.h - SPIRVBuiltIn to NVVM 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 SPIRVBuiltIn dialect to NVVM dialect.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef MLIR_CONVERSION_SPIRVBUILTINTONVVM_SPIRVBUILTINTONVVM_H
+#define MLIR_CONVERSION_SPIRVBUILTINTONVVM_SPIRVBUILTINTONVVM_H
+
+#include <memory>
+
+namespace mlir {
+class Pass;
+
+#define GEN_PASS_DECL_CONVERTSPIRVBUILTINTONVVM
+#include "mlir/Conversion/SYCLPasses.h.inc"
+#undef GEN_PASS_DECL_CONVERTSPIRVBUILTINTONVVM
+
+} // namespace mlir
+
+#endif // MLIR_CONVERSION_SPIRVBUILTINTONVVM_SPIRVBUILTINTONVVM_H

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

@@ -9,6 +9,7 @@
 #ifndef MLIR_CONVERSION_SYCLPASSES_H
 #define MLIR_CONVERSION_SYCLPASSES_H
 
+#include "mlir/Conversion/SPIRVBuiltInToNVVM/SPIRVBuiltInToNVVM.h"
 #include "mlir/Conversion/SYCLToGPU/SYCLToGPU.h"
 #include "mlir/Conversion/SYCLToLLVM/SYCLToLLVM.h"
 #include "mlir/Conversion/SYCLToMath/SYCLToMath.h"

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

@@ -121,4 +121,23 @@ def ConvertSYCLToMath : Pass<"convert-sycl-to-math", "ModuleOp"> {
   ];
 }
 
+//===----------------------------------------------------------------------===//
+// SPIRVBuiltInToNVVM
+//===----------------------------------------------------------------------===//
+
+def ConvertSPIRVBuiltInToNVVM : Pass<"convert-spirv-builtin-to-nvvm", "ModuleOp"> {
+  let summary = "convert __spirv_BuiltIn access to NVVM intrinsics";
+  let description = [{
+    Pass to support SYCL-MLIR running on nvidia gpus.
+
+    This Pass will be used to modify the final mlir which only contains llvm dialect.
+    It enables the final MLIR to be converted into LLVM IR that can run on NVIDIA GPUs, mainly by doing two things:
+    1.Mark the kernel function
+    2.Replaces SPIR-V builtin global variables and accessor logic with direct NVVM intrinsic calls.
+  }];
+  let dependentDialects = [
+      "LLVM::LLVMDialect"
+  ];
+}
+
 #endif // MLIR_CONVERSION_SYCLPASSES

mlir-sycl/lib/Conversion/CMakeLists.txt

@@ -2,3 +2,4 @@ add_subdirectory(SYCLToGPU)
 add_subdirectory(SYCLToLLVM)
 add_subdirectory(SYCLToMath)
 add_subdirectory(SYCLToSPIRV)
+add_subdirectory(SPIRVBuiltInToNVVM)

mlir-sycl/lib/Conversion/SPIRVBuiltInToNVVM/CMakeLists.txt

@@ -0,0 +1,17 @@
+add_mlir_conversion_library(MLIRSPIRVBuiltInToNVVM
+  SPIRVBuiltInToNVVM.cpp
+
+  ADDITIONAL_HEADER_DIRS
+  ${MLIR_SYCL_SOURCE_DIR}/mlir/Conversion/SPIRVBuiltInToNVVM
+
+  DEPENDS
+  MLIRSYCLConversionPassIncGen
+
+  LINK_LIBS PUBLIC
+  MLIRLLVMDialect
+  MLIRIR
+  MLIRSYCLDialect
+  MLIRTransforms
+  MLIRVectorDialect
+  )
+