[Mlir-commits] [mlir] [mlir][affine] Use `lookupNearestSymbolFrom` to simplify code (NFC) (PR #176679)

[Longsheng Mou]

https://github.com/CoTinker created https://github.com/llvm/llvm-project/pull/176679

None

>From 350ab61824e492d008fdc47371cd52170150a10f Mon Sep 17 00:00:00 2001
From: Longsheng Mou <longshengmou at gmail.com>
Date: Mon, 19 Jan 2026 00:48:56 +0800
Subject: [PATCH] [mlir][affine] Use `lookupNearestSymbolFrom` to simplify code
 (NFC)

---
 mlir/lib/Dialect/Affine/IR/AffineOps.cpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp
index 84813810cfa57..70e163e143c10 100644
--- a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp
+++ b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp
@@ -3478,11 +3478,8 @@ OpFoldResult AffineLoadOp::fold(FoldAdaptor adaptor) {
   if (!getGlobalOp)
     return {};
   // Get to the memref.global defining the symbol.
-  auto *symbolTableOp = getGlobalOp->getParentWithTrait<OpTrait::SymbolTable>();
-  if (!symbolTableOp)
-    return {};
-  auto global = dyn_cast_or_null<memref::GlobalOp>(
-      SymbolTable::lookupSymbolIn(symbolTableOp, getGlobalOp.getNameAttr()));
+  auto global = SymbolTable::lookupNearestSymbolFrom<memref::GlobalOp>(
+      getGlobalOp, getGlobalOp.getNameAttr());
   if (!global)
     return {};