From bd84a76001e6df033ec111ebc285c35d50a91342 Mon Sep 17 00:00:00 2001
From: Armin Ale <arminale@tenstorrent.com>
Date: Wed, 26 Feb 2025 20:18:19 +0000
Subject: [PATCH] replace auto with concrete type

---
 lib/Dialect/TTNN/IR/TTNNOpModelInterface.cpp | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/lib/Dialect/TTNN/IR/TTNNOpModelInterface.cpp b/lib/Dialect/TTNN/IR/TTNNOpModelInterface.cpp
index 0e9173b85a..d1874441f6 100644
--- a/lib/Dialect/TTNN/IR/TTNNOpModelInterface.cpp
+++ b/lib/Dialect/TTNN/IR/TTNNOpModelInterface.cpp
@@ -58,7 +58,7 @@ ReluOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto outputShape =
       mlir::cast<RankedTensorType>(getResults().front().getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }
@@ -100,7 +100,7 @@ AddOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto outputShape =
       mlir::cast<RankedTensorType>(getResult(0).getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }
@@ -141,7 +141,7 @@ SoftmaxOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto outputShape =
       mlir::cast<RankedTensorType>(getResult().getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }
@@ -177,7 +177,7 @@ MeanOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto inputShape =
       mlir::cast<RankedTensorType>(getOperand().getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }
@@ -215,7 +215,7 @@ ReshapeOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto outputShape =
       mlir::cast<RankedTensorType>(getResult().getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }
@@ -255,7 +255,7 @@ MatmulOp::getOpConstraints(const std::vector<TTNNLayoutAttr> &inputs,
   const auto outputShape =
       mlir::cast<RankedTensorType>(getResult().getType()).getShape();
 
-  auto check = detail::checkDeviceWorkerGrid(getOperation());
+  llvm::Expected<bool> check = detail::checkDeviceWorkerGrid(getOperation());
   if (!check) {
     return check.takeError();
   }