//===- FuncToSPIRVPass.cpp - Func to SPIR-V Passes ----------------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// // // This file implements a pass to convert Func dialect to SPIR-V dialect. // //===----------------------------------------------------------------------===// #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRVPass.h" #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRV.h" #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" #include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h" namespace mlir { #define GEN_PASS_DEF_CONVERTFUNCTOSPIRV #include "mlir/Conversion/Passes.h.inc" } // namespace mlir using namespace mlir; namespace { /// A pass converting MLIR Func operations into the SPIR-V dialect. class ConvertFuncToSPIRVPass : public impl::ConvertFuncToSPIRVBase { void runOnOperation() override; }; } // namespace void ConvertFuncToSPIRVPass::runOnOperation() { MLIRContext *context = &getContext(); Operation *op = getOperation(); auto targetAttr = spirv::lookupTargetEnvOrDefault(op); std::unique_ptr target = SPIRVConversionTarget::get(targetAttr); SPIRVConversionOptions options; options.emulateLT32BitScalarTypes = this->emulateLT32BitScalarTypes; SPIRVTypeConverter typeConverter(targetAttr, options); RewritePatternSet patterns(context); populateFuncToSPIRVPatterns(typeConverter, patterns); populateBuiltinFuncToSPIRVPatterns(typeConverter, patterns); if (failed(applyPartialConversion(op, *target, std::move(patterns)))) return signalPassFailure(); } std::unique_ptr> mlir::createConvertFuncToSPIRVPass() { return std::make_unique(); }