Put VPDT as pointer into State

Author: Mel-Chen

llvm/lib/Transforms/Vectorize/VPlan.cpp

@@ -222,7 +222,9 @@ VPTransformState::VPTransformState(const TargetTransformInfo *TTI,
                                    Loop *CurrentParentLoop, Type *CanonicalIVTy)
     : TTI(TTI), VF(VF), CFG(DT), LI(LI), Builder(Builder), ILV(ILV), Plan(Plan),
       CurrentParentLoop(CurrentParentLoop), LVer(nullptr),
-      TypeAnalysis(CanonicalIVTy) {}
+      TypeAnalysis(CanonicalIVTy), VPDT(new VPDominatorTree(*Plan)) {}
+
+VPTransformState::~VPTransformState() { delete VPDT; }
 
 Value *VPTransformState::get(VPValue *Def, const VPLane &Lane) {
   if (Def->isLiveIn())
@@ -265,11 +267,10 @@ Value *VPTransformState::get(VPValue *Def, bool NeedsScalar) {
     return Data.VPV2Vector[Def];
 
   auto GetBroadcastInstrs = [this, Def](Value *V) {
-    VPDominatorTree VPDT(*Plan);
     bool SafeToHoist =
         !Def->hasDefiningRecipe() ||
-        VPDT.properlyDominates(Def->getDefiningRecipe()->getParent(),
-                               Plan->getVectorPreheader());
+        VPDT->properlyDominates(Def->getDefiningRecipe()->getParent(),
+                                Plan->getVectorPreheader());
 
     if (VF.isScalar())
       return V;

llvm/lib/Transforms/Vectorize/VPlanHelpers.h

@@ -38,6 +38,7 @@ class VPRegionBlock;
 class VPlan;
 class Value;
 class LoopVersioning;
+class VPDominatorTree;
 
 /// Returns a calculation for the total number of elements for a given \p VF.
 /// For fixed width vectors this value is a constant, whereas for scalable
@@ -351,6 +352,9 @@ struct VPTransformState {
 
   /// VPlan-based type analysis.
   VPTypeAnalysis TypeAnalysis;
+
+  /// VPlan-based dominator tree.
+  VPDominatorTree *VPDT = nullptr;
 };
 
 /// Struct to hold various analysis needed for cost computations.