1 //===- FuncToSPIRVPass.cpp - Func 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 Func dialect to SPIR-V dialect. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRVPass.h" 14 15 #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRV.h" 16 #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" 17 #include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h" 18 19 namespace mlir { 20 #define GEN_PASS_DEF_CONVERTFUNCTOSPIRV 21 #include "mlir/Conversion/Passes.h.inc" 22 } // namespace mlir 23 24 using namespace mlir; 25 26 namespace { 27 /// A pass converting MLIR Func operations into the SPIR-V dialect. 28 class ConvertFuncToSPIRVPass 29 : public impl::ConvertFuncToSPIRVBase<ConvertFuncToSPIRVPass> { 30 void runOnOperation() override; 31 }; 32 } // namespace 33 runOnOperation()34void ConvertFuncToSPIRVPass::runOnOperation() { 35 MLIRContext *context = &getContext(); 36 Operation *op = getOperation(); 37 38 auto targetAttr = spirv::lookupTargetEnvOrDefault(op); 39 std::unique_ptr<ConversionTarget> target = 40 SPIRVConversionTarget::get(targetAttr); 41 42 SPIRVConversionOptions options; 43 options.emulateLT32BitScalarTypes = this->emulateLT32BitScalarTypes; 44 SPIRVTypeConverter typeConverter(targetAttr, options); 45 46 RewritePatternSet patterns(context); 47 populateFuncToSPIRVPatterns(typeConverter, patterns); 48 populateBuiltinFuncToSPIRVPatterns(typeConverter, patterns); 49 50 if (failed(applyPartialConversion(op, *target, std::move(patterns)))) 51 return signalPassFailure(); 52 } 53 createConvertFuncToSPIRVPass()54std::unique_ptr<OperationPass<>> mlir::createConvertFuncToSPIRVPass() { 55 return std::make_unique<ConvertFuncToSPIRVPass>(); 56 } 57