1//===- ControlFlowToSPIRVPass.cpp - ControlFlow to SPIR-V 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// This file implements a pass to convert ControlFlow dialect to SPIR-V dialect.
10//
11//===----------------------------------------------------------------------===//
12
13#include "mlir/Conversion/ControlFlowToSPIRV/ControlFlowToSPIRVPass.h"
14
15#include "mlir/Conversion/ControlFlowToSPIRV/ControlFlowToSPIRV.h"
16#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
17#include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h"
18
19namespace mlir {
20#define GEN_PASS_DEF_CONVERTCONTROLFLOWTOSPIRVPASS
21#include "mlir/Conversion/Passes.h.inc"
22} // namespace mlir
23
24using namespace mlir;
25
26namespace {
27/// A pass converting MLIR ControlFlow operations into the SPIR-V dialect.
28class ConvertControlFlowToSPIRVPass final
29 : public impl::ConvertControlFlowToSPIRVPassBase<
30 ConvertControlFlowToSPIRVPass> {
31 using Base::Base;
32 void runOnOperation() override;
33};
34} // namespace
35
36void ConvertControlFlowToSPIRVPass::runOnOperation() {
37 MLIRContext *context = &getContext();
38 Operation *op = getOperation();
39
40 auto targetAttr = spirv::lookupTargetEnvOrDefault(op);
41 std::unique_ptr<ConversionTarget> target =
42 SPIRVConversionTarget::get(targetAttr: targetAttr);
43
44 SPIRVConversionOptions options;
45 options.emulateLT32BitScalarTypes = this->emulateLT32BitScalarTypes;
46 SPIRVTypeConverter typeConverter(targetAttr, options);
47
48 // TODO: We should also take care of block argument type conversion.
49
50 RewritePatternSet patterns(context);
51 cf::populateControlFlowToSPIRVPatterns(typeConverter, patterns);
52
53 if (failed(applyPartialConversion(op, *target, std::move(patterns))))
54 return signalPassFailure();
55}
56

Provided by KDAB

Privacy Policy
Update your C++ knowledge – Modern C++11/14/17 Training
Find out more

source code of mlir/lib/Conversion/ControlFlowToSPIRV/ControlFlowToSPIRVPass.cpp