fixup! [loop count assumptions] convert loop iteration metadata to as…

…sumptions
Xilinx · Nov 26, 2024 · fa41660 · fa41660
1 parent 89dd3b2
commit fa41660
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/llvm/lib/Transforms/Utils/LoopIterCountAssumptions.cpp b/llvm/lib/Transforms/Utils/LoopIterCountAssumptions.cpp
@@ -131,11 +131,12 @@ void insertMinIterAssumption(ICmpInst &LoopCmpInstr, BasicBlock &LoopHeader,
   // expression can be simplified in later passes
   SCEV::NoWrapFlags NWF = BunldedSCEV.AddRecExpr->getNoWrapFlags(
       SCEV::NoWrapFlags(/*Mask=*/SCEV::FlagNUW | SCEV::FlagNSW));
-  SCEVCommutativeExpr *AE =
-      dyn_cast<SCEVCommutativeExpr>(const_cast<SCEV *>(MinIterSCEV));
-  if (AE && NWF) {
-    AE->setNoWrapFlags(NWF);
-    MinIterSCEV = dyn_cast<const SCEV>(AE);
+  const SCEVCommutativeExpr *ConstCE =
+      dyn_cast<SCEVCommutativeExpr>(MinIterSCEV);
+  if (ConstCE && NWF) {
+    SCEVCommutativeExpr *CE = const_cast<SCEVCommutativeExpr *>(ConstCE);
+    CE->setNoWrapFlags(NWF);
+    MinIterSCEV = dyn_cast<const SCEV>(CE);
   }
 
   // Expansion of MinIterSCEV will result in                                   \