================
@@ -5801,23 +5806,30 @@ void VPlanTransforms::expandSCEVsToVPInstructions(VPlan 
&Plan,
                     ->getDebugLoc();
   VPSCEVExpander Expander(Builder, SE, DL);
 
-  // Expand VPExpandSCEVRecipes to VPInstructions using VPSCEVExpander. During
-  // the transition, unsupported VPExpandSCEVRecipes are skipped and left for
-  // late expansion.
+  // Expand VPExpandSCEVRecipes and VPExpandStridePredicatesRecipes to
+  // VPInstructions using VPSCEVExpander. During the transition, unsupported
+  // recipes are skipped and left for late expansion.
   for (VPRecipeBase &R : make_early_inc_range(*Entry)) {
-    auto *ExpSCEV = dyn_cast<VPExpandSCEVRecipe>(&R);
-    if (!ExpSCEV || ExpSCEV->user_empty())
+    auto *Def = dyn_cast<VPSingleDefRecipe>(&R);
+    if (!Def || Def->user_empty())
+      continue;
----------------
artagnon wrote:

```suggestion
    if (!isa<VPExpandStridePredicatesRecipe, VPExpandSCEVRecipe>(&R) || 
R.user_empty())
      continue;
```

https://github.com/llvm/llvm-project/pull/182595
_______________________________________________
llvm-branch-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits

Reply via email to