[Mlir-commits] [mlir] [Tosa] Fix TosaValidation for FuncOp (PR #69997)
Mehdi Amini
llvmlistbot at llvm.org
Mon Oct 23 22:17:43 PDT 2023
================
@@ -0,0 +1,66 @@
+//===- TosaToLinalgPipeline.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 "mlir/Conversion/TosaToLinalg/TosaToLinalg.h"
+
+#include "mlir/Dialect/Arith/IR/Arith.h"
+#include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/Dialect/Linalg/IR/Linalg.h"
+#include "mlir/Dialect/Math/IR/Math.h"
+#include "mlir/Dialect/SCF/IR/SCF.h"
+#include "mlir/Dialect/Tensor/IR/Tensor.h"
+#include "mlir/Dialect/Tosa/IR/TosaOps.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"
+
+namespace mlir {
+#define GEN_PASS_DEF_TOSATOLINALGPIPELINE
+#include "mlir/Conversion/Passes.h.inc"
+} // namespace mlir
+
+using namespace mlir;
+
+namespace {
+struct TosaToLinalgPipeline
+ : public impl::TosaToLinalgPipelineBase<TosaToLinalgPipeline> {
+public:
+ void getDependentDialects(DialectRegistry ®istry) const override {
+ registry
+ .insert<arith::ArithDialect, linalg::LinalgDialect, math::MathDialect,
+ tensor::TensorDialect, scf::SCFDialect>();
+ }
+
+ void runOnOperation() override {
+ OpPassManager pm("builtin.module");
+
+ TosaToLinalgOptions tosaToLinalgOptions;
+
+ tosa::addTosaToLinalgPasses(pm, tosaToLinalgOptions,
+ /* validationOptions = */
+ {tosa::TosaProfileEnum::BaseInference,
+ /* StrictOperationSpecAlignment = */ true,
+ tosa::TosaLevelEnum::EightK});
+
+ if (failed(runPipeline(pm, getOperation())))
+ signalPassFailure();
+ }
----------------
joker-eph wrote:
Why isn't this just a registered pipeline instead of a pass? https://mlir.llvm.org/docs/PassManagement/#pass-pipeline-registration
https://github.com/llvm/llvm-project/pull/69997
More information about the Mlir-commits
mailing list