From b8ad68fcc3c453728f4b91ab0c24a3866b59a3c9 Mon Sep 17 00:00:00 2001
From: Kunwar Grover <groverkss@gmail.com>
Date: Fri, 20 Oct 2023 12:35:30 +0530
Subject: [PATCH] [mlir][tosa] Update pass pipeline for TosaToLinalg (#69679)

This patch fixes the nesting of TosaValidation pass added in
TosaToLinalg pipeline.
---
 mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp
index 3c54f85b033b0b..2072fabc29242e 100644
--- 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());
 }