mlir/lib/Conversion/SPIRVToLLVM/SPIRVToLLVMPass.cpp

   1 //===- SPIRVToLLVMPass.cpp - SPIR-V to LLVM Passes ------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file implements a pass to convert MLIR SPIR-V ops into LLVM ops
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "mlir/Conversion/SPIRVToLLVM/SPIRVToLLVMPass.h"
  14
  15 #include "mlir/Conversion/LLVMCommon/TypeConverter.h"
  16 #include "mlir/Conversion/SPIRVToLLVM/SPIRVToLLVM.h"
  17 #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
  18 #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
  19 #include "mlir/Dialect/SPIRV/IR/SPIRVEnums.h"
  20 #include "mlir/Pass/Pass.h"
  21
  22 namespace mlir {
  23 #define GEN_PASS_DEF_CONVERTSPIRVTOLLVMPASS
  24 #include "mlir/Conversion/Passes.h.inc"
  25 } // namespace mlir
  26
  27 using namespace mlir;
  28
  29 namespace {
  30 /// A pass converting MLIR SPIR-V operations into LLVM dialect.
  31 class ConvertSPIRVToLLVMPass
  32     : public impl::ConvertSPIRVToLLVMPassBase<ConvertSPIRVToLLVMPass> {
  33   void runOnOperation() override;
  34
  35 public:
  36   using Base::Base;
  37 };
  38 } // namespace
  39
  40 void ConvertSPIRVToLLVMPass::runOnOperation() {
  41   MLIRContext *context = &getContext();
  42   ModuleOp module = getOperation();
  43
  44   LowerToLLVMOptions options(&getContext());
  45
  46   LLVMTypeConverter converter(&getContext(), options);
  47
  48   // Encode global variable's descriptor set and binding if they exist.
  49   encodeBindAttribute(module);
  50
  51   RewritePatternSet patterns(context);
  52
  53   populateSPIRVToLLVMTypeConversion(converter, clientAPI);
  54
  55   populateSPIRVToLLVMModuleConversionPatterns(converter, patterns);
  56   populateSPIRVToLLVMConversionPatterns(converter, patterns, clientAPI);
  57   populateSPIRVToLLVMFunctionConversionPatterns(converter, patterns);
  58
  59   ConversionTarget target(*context);
  60   target.addIllegalDialect<spirv::SPIRVDialect>();
  61   target.addLegalDialect<LLVM::LLVMDialect>();
  62
  63   if (clientAPI != spirv::ClientAPI::OpenCL &&
  64       clientAPI != spirv::ClientAPI::Unknown)
  65     getOperation()->emitWarning()
  66         << "address space mapping for client '"
  67         << spirv::stringifyClientAPI(clientAPI) << "' not implemented";
  68
  69   // Set `ModuleOp` as legal for `spirv.module` conversion.
  70   target.addLegalOp<ModuleOp>();
  71   if (failed(applyPartialConversion(module, target, std::move(patterns))))
  72     signalPassFailure();
  73 }