//===- SCFToSPIRVPass.cpp - SCF 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 SCF dialect into SPIR-V dialect. // //===----------------------------------------------------------------------===// #include "mlir/Conversion/SCFToSPIRV/SCFToSPIRVPass.h" #include "mlir/Conversion/ArithToSPIRV/ArithToSPIRV.h" #include "mlir/Conversion/FuncToSPIRV/FuncToSPIRV.h" #include "mlir/Conversion/IndexToSPIRV/IndexToSPIRV.h" #include "mlir/Conversion/MemRefToSPIRV/MemRefToSPIRV.h" #include "mlir/Conversion/SCFToSPIRV/SCFToSPIRV.h" #include "mlir/Dialect/SCF/IR/SCF.h" #include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h" #include "mlir/Dialect/SPIRV/Transforms/SPIRVConversion.h" namespace mlir { #define GEN_PASS_DEF_SCFTOSPIRV #include "mlir/Conversion/Passes.h.inc" } // namespace mlir using namespace mlir; namespace { struct SCFToSPIRVPass : public impl::SCFToSPIRVBase { void runOnOperation() override; }; } // namespace void SCFToSPIRVPass::runOnOperation() { MLIRContext *context = &getContext(); Operation *op = getOperation(); auto targetAttr = spirv::lookupTargetEnvOrDefault(op); std::unique_ptr target = SPIRVConversionTarget::get(targetAttr); SPIRVTypeConverter typeConverter(targetAttr); ScfToSPIRVContext scfContext; RewritePatternSet patterns(context); populateSCFToSPIRVPatterns(typeConverter, scfContext, patterns); // TODO: Change SPIR-V conversion to be progressive and remove the following // patterns. mlir::arith::populateArithToSPIRVPatterns(typeConverter, patterns); populateFuncToSPIRVPatterns(typeConverter, patterns); populateMemRefToSPIRVPatterns(typeConverter, patterns); populateBuiltinFuncToSPIRVPatterns(typeConverter, patterns); index::populateIndexToSPIRVPatterns(typeConverter, patterns); if (failed(applyPartialConversion(op, *target, std::move(patterns)))) return signalPassFailure(); }