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 #include "../PassDetail.h" 15 #include "mlir/Conversion/ControlFlowToSPIRV/ControlFlowToSPIRV.h" 16 #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" 17 #include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h" 18 19 using namespace mlir; 20 21 namespace { 22 /// A pass converting MLIR ControlFlow operations into the SPIR-V dialect. 23 class ConvertControlFlowToSPIRVPass 24 : public ConvertControlFlowToSPIRVBase<ConvertControlFlowToSPIRVPass> { 25 void runOnOperation() override; 26 }; 27 } // namespace 28 runOnOperation()29void ConvertControlFlowToSPIRVPass::runOnOperation() { 30 MLIRContext *context = &getContext(); 31 ModuleOp module = getOperation(); 32 33 auto targetAttr = spirv::lookupTargetEnvOrDefault(module); 34 std::unique_ptr<ConversionTarget> target = 35 SPIRVConversionTarget::get(targetAttr); 36 37 SPIRVTypeConverter::Options options; 38 options.emulateNon32BitScalarTypes = this->emulateNon32BitScalarTypes; 39 SPIRVTypeConverter typeConverter(targetAttr, options); 40 41 RewritePatternSet patterns(context); 42 cf::populateControlFlowToSPIRVPatterns(typeConverter, patterns); 43 44 if (failed(applyPartialConversion(module, *target, std::move(patterns)))) 45 return signalPassFailure(); 46 } 47 48 std::unique_ptr<OperationPass<ModuleOp>> createConvertControlFlowToSPIRVPass()49mlir::createConvertControlFlowToSPIRVPass() { 50 return std::make_unique<ConvertControlFlowToSPIRVPass>(); 51 } 52