From b5b140d4c789911e972d4f2ff4e1c718184f5a9c Mon Sep 17 00:00:00 2001
From: Filip Bajraktari <fbajraktari@tenstorrent.com>
Date: Thu, 12 Dec 2024 11:16:21 +0000
Subject: [PATCH] fix comment for TTNNLayoutAttr::getShardShape()

---
 lib/Dialect/TTNN/IR/TTNNOpsAttrs.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/Dialect/TTNN/IR/TTNNOpsAttrs.cpp b/lib/Dialect/TTNN/IR/TTNNOpsAttrs.cpp
index f62f6c0293..dbefdf3473 100644
--- a/lib/Dialect/TTNN/IR/TTNNOpsAttrs.cpp
+++ b/lib/Dialect/TTNN/IR/TTNNOpsAttrs.cpp
@@ -177,7 +177,7 @@ uint64_t TTNNLayoutAttr::getElementSizeBytes() const {
 // Return the shape of the shard.
 // Example: memref<2x2x!tt.tile<32x32xf32>> -> { 2, 2 }
 // Example: memref<128x128xf32> -> { 128, 128 }
-// Example: memref<2x3!tt.tile<32x32xf32>> -> { 2, 3 }
+// Example: memref<2x3x!tt.tile<32x32xf32>> -> { 2, 3 }
 //
 // return The shape of the shard.
 llvm::SmallVector<int64_t> TTNNLayoutAttr::getShardShape() const {