[llvm] 8fb802e - [LV] Improve code in collectInstsToScalarize (NFC) (#130643)

[via llvm-commits]

Author: Ramkumar Ramachandra
Date: 2025-03-25T16:52:13Z
New Revision: 8fb802e995868f19f40ff7a14df6f38b248af3ce

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

LOG: [LV] Improve code in collectInstsToScalarize (NFC) (#130643)

Added: 
    

Modified: 
    llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
index 67509bd598f6a..18be2c0ce8c2e 100644
--- a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
+++ b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
@@ -5391,12 +5391,12 @@ void LoopVectorizationCostModel::collectInstsToScalarize(ElementCount VF) {
           ScalarCostsVF.insert(ScalarCosts.begin(), ScalarCosts.end());
           // Check if we decided to scalarize a call. If so, update the widening
           // decision of the call to CM_Scalarize with the computed scalar cost.
-          for (const auto &[I, _] : ScalarCosts) {
+          for (const auto &[I, Cost] : ScalarCosts) {
             auto *CI = dyn_cast<CallInst>(I);
             if (!CI || !CallWideningDecisions.contains({CI, VF}))
               continue;
             CallWideningDecisions[{CI, VF}].Kind = CM_Scalarize;
-            CallWideningDecisions[{CI, VF}].Cost = ScalarCosts[CI];
+            CallWideningDecisions[{CI, VF}].Cost = Cost;
           }
         }
         // Remember that BB will remain after vectorization.