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

[llvmlistbot at llvm.org]

Author: Longsheng Mou
Date: 2026-01-19T19:03:01+08:00
New Revision: be037864f969ca47c61d17e1d1c65cb21477650a

URL: https://github.com/llvm/llvm-project/commit/be037864f969ca47c61d17e1d1c65cb21477650a
DIFF: https://github.com/llvm/llvm-project/commit/be037864f969ca47c61d17e1d1c65cb21477650a.diff

LOG: [mlir][affine] Use `lookupNearestSymbolFrom` to simplify code (NFC) (#176679)

Added: 
    

Modified: 
    mlir/lib/Dialect/Affine/IR/AffineOps.cpp

Removed: 
    


################################################################################
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 {};