Put VPDT as pointer into State

Author: Mel-Chen

llvm/lib/Transforms/Vectorize/VPlan.cpp

@@ -220,7 +220,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())
@@ -263,11 +265,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/VPlan.h

@@ -67,6 +67,7 @@ class VPlanSlp;
 class Value;
 class LoopVectorizationCostModel;
 class LoopVersioning;
+class VPDominatorTree;
 
 struct VPCostContext;
 
@@ -238,6 +239,7 @@ struct VPTransformState {
                    LoopInfo *LI, DominatorTree *DT, IRBuilderBase &Builder,
                    InnerLoopVectorizer *ILV, VPlan *Plan,
                    Loop *CurrentParentLoop, Type *CanonicalIVTy);
+  ~VPTransformState();
   /// Target Transform Info.
   const TargetTransformInfo *TTI;
 
@@ -390,6 +392,9 @@ struct VPTransformState {
 
   /// VPlan-based type analysis.
   VPTypeAnalysis TypeAnalysis;
+
+  /// VPlan-based dominator tree.
+  VPDominatorTree *VPDT = nullptr;
 };
 
 /// VPBlockBase is the building block of the Hierarchical Control-Flow Graph.