47f175b09b
These passes generally don't rely on any special aspects of FuncOp, and moving allows for these passes to be used in many more situations. The passes that obviously weren't relying on invariants guaranteed by a "function" were updated to be generic pass, the rest were updated to be FunctionOpinterface InterfacePasses. The test updates are NFC switching from implicit nesting (-pass -pass2) form to the -pass-pipeline form (generic passes do not implicitly nest as op-specific passes do). Differential Revision: https://reviews.llvm.org/D121190
82 lines
3.1 KiB
C++
82 lines
3.1 KiB
C++
//===- TosaToLinalgPass.cpp - Lowering Tosa to Linalg Dialect -------------===//
|
|
//
|
|
// 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 transformation pass legalizes Tosa operations to the Linalg dialect.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "../PassDetail.h"
|
|
#include "mlir/Conversion/TosaToLinalg/TosaToLinalg.h"
|
|
#include "mlir/Dialect/Arithmetic/IR/Arithmetic.h"
|
|
#include "mlir/Dialect/Linalg/IR/Linalg.h"
|
|
#include "mlir/Dialect/Math/IR/Math.h"
|
|
#include "mlir/Dialect/SCF/SCF.h"
|
|
#include "mlir/Dialect/Tensor/IR/Tensor.h"
|
|
#include "mlir/Dialect/Tosa/IR/TosaOps.h"
|
|
#include "mlir/Dialect/Tosa/Transforms/PassDetail.h"
|
|
#include "mlir/Dialect/Tosa/Transforms/Passes.h"
|
|
#include "mlir/Dialect/Tosa/Utils/QuantUtils.h"
|
|
#include "mlir/IR/PatternMatch.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Transforms/DialectConversion.h"
|
|
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
struct TosaToLinalg : public TosaToLinalgBase<TosaToLinalg> {
|
|
public:
|
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
|
registry
|
|
.insert<arith::ArithmeticDialect, linalg::LinalgDialect,
|
|
math::MathDialect, tensor::TensorDialect, scf::SCFDialect>();
|
|
}
|
|
|
|
void runOnOperation() override {
|
|
RewritePatternSet patterns(&getContext());
|
|
ConversionTarget target(getContext());
|
|
target.addLegalDialect<linalg::LinalgDialect, tensor::TensorDialect,
|
|
scf::SCFDialect>();
|
|
target.addIllegalDialect<tosa::TosaDialect>();
|
|
|
|
// Not every TOSA op can be legalized to linalg.
|
|
target.addLegalOp<tosa::ApplyScaleOp>();
|
|
target.addLegalOp<tosa::IfOp>();
|
|
target.addLegalOp<tosa::ConstOp>();
|
|
target.addLegalOp<tosa::WhileOp>();
|
|
target.addLegalOp<tosa::SliceOp>();
|
|
|
|
target.markUnknownOpDynamicallyLegal([](Operation *) { return true; });
|
|
|
|
FunctionOpInterface func = getOperation();
|
|
mlir::tosa::populateTosaToLinalgConversionPatterns(&patterns);
|
|
if (failed(applyFullConversion(func, target, std::move(patterns))))
|
|
signalPassFailure();
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
std::unique_ptr<Pass> mlir::tosa::createTosaToLinalg() {
|
|
return std::make_unique<TosaToLinalg>();
|
|
}
|
|
|
|
void mlir::tosa::addTosaToLinalgPasses(OpPassManager &pm,
|
|
bool disableTosaDecompositions) {
|
|
// Optional decompositions are designed to benefit linalg.
|
|
if (!disableTosaDecompositions)
|
|
pm.addNestedPass<FuncOp>(mlir::tosa::createTosaOptionalDecompositions());
|
|
pm.addNestedPass<FuncOp>(mlir::createCanonicalizerPass());
|
|
|
|
pm.addNestedPass<FuncOp>(tosa::createTosaMakeBroadcastablePass());
|
|
pm.addNestedPass<FuncOp>(tosa::createTosaToLinalgNamed());
|
|
pm.addNestedPass<FuncOp>(mlir::createCanonicalizerPass());
|
|
pm.addNestedPass<FuncOp>(tosa::createTosaMakeBroadcastablePass());
|
|
pm.addNestedPass<FuncOp>(tosa::createTosaToLinalg());
|
|
}
|