1 //===- ConvertStandardToSPIRVPass.cpp - Convert Std Ops to SPIR-V Ops -----===// 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 MLIR standard ops into the SPIR-V 10 // ops. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "mlir/Conversion/StandardToSPIRV/ConvertStandardToSPIRVPass.h" 15 #include "../PassDetail.h" 16 #include "mlir/Conversion/StandardToSPIRV/ConvertStandardToSPIRV.h" 17 #include "mlir/Dialect/SPIRV/SPIRVDialect.h" 18 #include "mlir/Dialect/SPIRV/SPIRVLowering.h" 19 20 using namespace mlir; 21 22 namespace { 23 /// A pass converting MLIR Standard operations into the SPIR-V dialect. 24 class ConvertStandardToSPIRVPass 25 : public ConvertStandardToSPIRVBase<ConvertStandardToSPIRVPass> { 26 void runOnOperation() override; 27 }; 28 } // namespace 29 runOnOperation()30void ConvertStandardToSPIRVPass::runOnOperation() { 31 MLIRContext *context = &getContext(); 32 ModuleOp module = getOperation(); 33 34 auto targetAttr = spirv::lookupTargetEnvOrDefault(module); 35 std::unique_ptr<ConversionTarget> target = 36 spirv::SPIRVConversionTarget::get(targetAttr); 37 38 SPIRVTypeConverter typeConverter(targetAttr); 39 OwningRewritePatternList patterns; 40 populateStandardToSPIRVPatterns(context, typeConverter, patterns); 41 populateBuiltinFuncToSPIRVPatterns(context, typeConverter, patterns); 42 43 if (failed(applyPartialConversion(module, *target, std::move(patterns)))) 44 return signalPassFailure(); 45 } 46 47 std::unique_ptr<OperationPass<ModuleOp>> createConvertStandardToSPIRVPass()48mlir::createConvertStandardToSPIRVPass() { 49 return std::make_unique<ConvertStandardToSPIRVPass>(); 50 } 51