From 0ec78636f089d636ecee19edb214404c24dfce5d Mon Sep 17 00:00:00 2001
From: Bill Varcho <varcho@google.com>
Date: Thu, 8 Oct 2026 14:16:13 -0700
Subject: [PATCH] [SDY][optimization] Avoid redundant SymbolTable mesh lookups
 in getCommonMeshOrRef.

PiperOrigin-RevId: 996066154
---
 shardy/dialect/sdy/ir/utils.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/shardy/dialect/sdy/ir/utils.cc b/shardy/dialect/sdy/ir/utils.cc
index ca77dca8e..f6623986d 100644
--- a/shardy/dialect/sdy/ir/utils.cc
+++ b/shardy/dialect/sdy/ir/utils.cc
@@ -255,7 +255,7 @@ Attribute getCommonMeshOrRef(ArrayRef<TensorShardingAttr> operandShardings,
   MeshAttr mesh;
   for (TensorShardingAttr sharding : llvm::concat<const TensorShardingAttr>(
            operandShardings, resultsShardings)) {
-    if (!sharding) {
+    if (!sharding || sharding.getMeshOrRef() == meshOrRef) {
       continue;
     }
     MeshAttr otherMesh = sharding.getMesh(symbolTable);