1 //===- LinalgToSPIRVPass.cpp - Linalg to SPIR-V conversion pass -----------===//
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 #include "mlir/Conversion/LinalgToSPIRV/LinalgToSPIRVPass.h"
10 #include "mlir/Conversion/LinalgToSPIRV/LinalgToSPIRV.h"
11 #include "mlir/Dialect/SPIRV/SPIRVDialect.h"
12 #include "mlir/Dialect/SPIRV/SPIRVLowering.h"
13 #include "mlir/Pass/Pass.h"
14 
15 using namespace mlir;
16 
17 namespace {
18 /// A pass converting MLIR Linalg ops into SPIR-V ops.
19 class LinalgToSPIRVPass : public ModulePass<LinalgToSPIRVPass> {
20 /// Include the generated pass utilities.
21 #define GEN_PASS_ConvertLinalgToSPIRV
22 #include "mlir/Conversion/Passes.h.inc"
23 
24   void runOnModule() override;
25 };
26 } // namespace
27 
28 void LinalgToSPIRVPass::runOnModule() {
29   MLIRContext *context = &getContext();
30   ModuleOp module = getModule();
31 
32   auto targetAttr = spirv::lookupTargetEnvOrDefault(module);
33   std::unique_ptr<ConversionTarget> target =
34       spirv::SPIRVConversionTarget::get(targetAttr);
35 
36   SPIRVTypeConverter typeConverter(targetAttr);
37   OwningRewritePatternList patterns;
38   populateLinalgToSPIRVPatterns(context, typeConverter, patterns);
39   populateBuiltinFuncToSPIRVPatterns(context, typeConverter, patterns);
40 
41   // Allow builtin ops.
42   target->addLegalOp<ModuleOp, ModuleTerminatorOp>();
43   target->addDynamicallyLegalOp<FuncOp>(
44       [&](FuncOp op) { return typeConverter.isSignatureLegal(op.getType()); });
45 
46   if (failed(applyFullConversion(module, *target, patterns)))
47     return signalPassFailure();
48 }
49 
50 std::unique_ptr<OpPassBase<ModuleOp>> mlir::createLinalgToSPIRVPass() {
51   return std::make_unique<LinalgToSPIRVPass>();
52 }
53