Move the code in planContainsAdditionalSimplifications

Author: Mel-Chen

llvm/lib/Transforms/Vectorize/LoopVectorize.cpp

@@ -7035,6 +7035,20 @@ static bool planContainsAdditionalSimplifications(VPlan &Plan,
         if (AddrI && vputils::isSingleScalar(WidenMemR->getAddr()) !=
                          CostCtx.isLegacyUniformAfterVectorization(AddrI, VF))
           return true;
+
+        if (WidenMemR->isReverse()) {
+          // If the stored value of a reverse store is invariant, LICM will
+          // hoist the reverse operation to the preheader. In this case, the
+          // result of the VPlan-based cost model will diverge from that of
+          // the legacy model.
+          if (auto *StoreR = dyn_cast<VPWidenStoreRecipe>(WidenMemR))
+            if (StoreR->getStoredValue()->isDefinedOutsideLoopRegions())
+              return true;
+
+          if (auto *StoreR = dyn_cast<VPWidenStoreEVLRecipe>(WidenMemR))
+            if (StoreR->getStoredValue()->isDefinedOutsideLoopRegions())
+              return true;
+        }
       }
 
       /// If a VPlan transform folded a recipe to one producing a single-scalar,
@@ -7055,23 +7069,6 @@ static bool planContainsAdditionalSimplifications(VPlan &Plan,
             cast<VPRecipeWithIRFlags>(R).getPredicate() !=
                 cast<CmpInst>(UI)->getPredicate())
           return true;
-
-        if (auto *MemR = dyn_cast<VPWidenMemoryRecipe>(&R)) {
-          if (MemR->isReverse()) {
-            // If the stored value of a reverse store is invariant, LICM will
-            // hoist the reverse operation to the preheader. In this case, the
-            // result of the VPlan-based cost model will diverge from that of
-            // the legacy model.
-            if (auto *StoreR = dyn_cast<VPWidenStoreRecipe>(MemR))
-              if (StoreR->getStoredValue()->isDefinedOutsideLoopRegions())
-                return true;
-
-            if (auto *StoreR = dyn_cast<VPWidenStoreEVLRecipe>(MemR))
-              if (StoreR->getStoredValue()->isDefinedOutsideLoopRegions())
-                return true;
-          }
-        }
-
         SeenInstrs.insert(UI);
       }
     }