[mlir][tosa] Update pass pipeline for TosaToLinalg (llvm#69679)

This patch fixes the nesting of TosaValidation pass added in TosaToLinalg pipeline.
madhur13490 · Oct 20, 2023 · b8ad68f · b8ad68f
1 parent 291d8ab
commit b8ad68f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp
@@ -90,6 +90,7 @@ void mlir::tosa::addTosaToLinalgPasses(
   pm.addNestedPass<func::FuncOp>(tosa::createTosaLayerwiseConstantFoldPass(
       {options.aggressiveReduceConstant}));
   pm.addNestedPass<func::FuncOp>(tosa::createTosaMakeBroadcastablePass());
-  pm.addNestedPass<func::FuncOp>(tosa::createTosaValidation(validationOptions));
+  pm.addNestedPass<mlir::ModuleOp>(
+      tosa::createTosaValidation(validationOptions));
   pm.addNestedPass<func::FuncOp>(tosa::createTosaToLinalg());
 }