[llvm] ec880b1 - [X86] Avoid repeated map lookups (NFC) (#127374)

[via llvm-commits]

Author: Kazu Hirata
Date: 2025-02-16T08:16:30-08:00
New Revision: ec880b1450c5b9526d6310d1a66cf3a5297551de

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

LOG: [X86] Avoid repeated map lookups (NFC) (#127374)

Added: 
    

Modified: 
    llvm/lib/Target/X86/X86LowerAMXType.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/X86/X86LowerAMXType.cpp b/llvm/lib/Target/X86/X86LowerAMXType.cpp
index cb6127fb85749..2ea78d44d67cf 100644
--- a/llvm/lib/Target/X86/X86LowerAMXType.cpp
+++ b/llvm/lib/Target/X86/X86LowerAMXType.cpp
@@ -190,8 +190,8 @@ Value *ShapeCalculator::getRowFromCol(Instruction *II, Value *V,
 
 Value *ShapeCalculator::getColFromRow(Instruction *II, Value *V,
                                       unsigned Granularity) {
-  if (Row2Col.count(V))
-    return Row2Col[V];
+  if (auto It = Row2Col.find(V); It != Row2Col.end())
+    return It->second;
   IRBuilder<> Builder(II);
   Value *RealCol = nullptr;
   if (isa<ConstantInt>(V))