Use recursivelyDeleteDeadRecipes to clear header masks

lukel97 · lukel97 · commit 0e261f36101d · 2024-12-18T01:48:39.000+08:00
diff --git a/llvm/lib/Transforms/Vectorize/VPlanTransforms.cpp b/llvm/lib/Transforms/Vectorize/VPlanTransforms.cpp
@@ -1573,11 +1573,12 @@ static void transformRecipestoEVLRecipes(VPlan &Plan, VPValue &EVL) {
     }
   }
 
-  for (VPRecipeBase *R : ToErase)
+  for (VPRecipeBase *R : reverse(ToErase)) {
+    SmallVector<VPValue *> PossiblyDead(R->operands());
     R->eraseFromParent();
-
-  for (VPValue *HeaderMask : collectAllHeaderMasks(Plan))
-    recursivelyDeleteDeadRecipes(HeaderMask);
+    for (VPValue *Op : PossiblyDead)
+      recursivelyDeleteDeadRecipes(Op);
+  }
 }
 
 /// Add a VPEVLBasedIVPHIRecipe and related recipes to \p Plan and

Original file line number	Diff line number	Diff line change
`@@ -1573,11 +1573,12 @@ static void transformRecipestoEVLRecipes(VPlan &Plan, VPValue &EVL) {`
`1573`	`1573`	`}`
`1574`	`1574`	`}`
`1575`	`1575`
`1576`		`- for (VPRecipeBase *R : ToErase)`
	`1576`	`+ for (VPRecipeBase *R : reverse(ToErase)) {`
	`1577`	`+ SmallVector<VPValue *> PossiblyDead(R->operands());`
`1577`	`1578`	`R->eraseFromParent();`
`1578`		`-`
`1579`		`- for (VPValue *HeaderMask : collectAllHeaderMasks(Plan))`
`1580`		`- recursivelyDeleteDeadRecipes(HeaderMask);`
	`1579`	`+ for (VPValue *Op : PossiblyDead)`
	`1580`	`+ recursivelyDeleteDeadRecipes(Op);`
	`1581`	`+ }`
`1581`	`1582`	`}`
`1582`	`1583`
`1583`	`1584`	`/// Add a VPEVLBasedIVPHIRecipe and related recipes to \p Plan and`