1 | //===- SCFToSPIRVPass.cpp - SCF to SPIR-V 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 SCF dialect into SPIR-V dialect. |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | |
13 | #include "mlir/Conversion/SCFToSPIRV/SCFToSPIRVPass.h" |
14 | |
15 | #include "mlir/Conversion/ArithToSPIRV/ArithToSPIRV.h" |
16 | #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRV.h" |
17 | #include "mlir/Conversion/IndexToSPIRV/IndexToSPIRV.h" |
18 | #include "mlir/Conversion/MemRefToSPIRV/MemRefToSPIRV.h" |
19 | #include "mlir/Conversion/SCFToSPIRV/SCFToSPIRV.h" |
20 | #include "mlir/Dialect/SCF/IR/SCF.h" |
21 | #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" |
22 | #include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h" |
23 | |
24 | namespace mlir { |
25 | #define GEN_PASS_DEF_SCFTOSPIRV |
26 | #include "mlir/Conversion/Passes.h.inc" |
27 | } // namespace mlir |
28 | |
29 | using namespace mlir; |
30 | |
31 | namespace { |
32 | struct SCFToSPIRVPass : public impl::SCFToSPIRVBase<SCFToSPIRVPass> { |
33 | void runOnOperation() override; |
34 | }; |
35 | } // namespace |
36 | |
37 | void SCFToSPIRVPass::runOnOperation() { |
38 | MLIRContext *context = &getContext(); |
39 | Operation *op = getOperation(); |
40 | |
41 | auto targetAttr = spirv::lookupTargetEnvOrDefault(op); |
42 | std::unique_ptr<ConversionTarget> target = |
43 | SPIRVConversionTarget::get(targetAttr: targetAttr); |
44 | |
45 | SPIRVTypeConverter typeConverter(targetAttr); |
46 | ScfToSPIRVContext scfContext; |
47 | RewritePatternSet patterns(context); |
48 | populateSCFToSPIRVPatterns(typeConverter, scfToSPIRVContext&: scfContext, patterns); |
49 | |
50 | // TODO: Change SPIR-V conversion to be progressive and remove the following |
51 | // patterns. |
52 | mlir::arith::populateArithToSPIRVPatterns(typeConverter, patterns); |
53 | populateFuncToSPIRVPatterns(typeConverter, patterns); |
54 | populateMemRefToSPIRVPatterns(typeConverter, patterns); |
55 | populateBuiltinFuncToSPIRVPatterns(typeConverter, patterns); |
56 | index::populateIndexToSPIRVPatterns(converter&: typeConverter, patterns); |
57 | |
58 | if (failed(applyPartialConversion(op, *target, std::move(patterns)))) |
59 | return signalPassFailure(); |
60 | } |
61 | |