llvm/lib/Transforms/Vectorize/VPlanDominatorTree.h

   1 //===-- VPlanDominatorTree.h ------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 ///
   9 /// \file
  10 /// This file implements dominator tree analysis for a single level of a VPlan's
  11 /// H-CFG.
  12 ///
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_TRANSFORMS_VECTORIZE_VPLANDOMINATORTREE_H
  16 #define LLVM_TRANSFORMS_VECTORIZE_VPLANDOMINATORTREE_H
  17
  18 #include "VPlan.h"
  19 #include "VPlanCFG.h"
  20 #include "llvm/ADT/GraphTraits.h"
  21 #include "llvm/IR/Dominators.h"
  22 #include "llvm/Support/GenericDomTree.h"
  23
  24 namespace llvm {
  25
  26 template <> struct DomTreeNodeTraits<VPBlockBase> {
  27   using NodeType = VPBlockBase;
  28   using NodePtr = VPBlockBase *;
  29   using ParentPtr = VPlan *;
  30
  31   static NodePtr getEntryNode(ParentPtr Parent) { return Parent->getEntry(); }
  32   static ParentPtr getParent(NodePtr B) { return B->getPlan(); }
  33 };
  34
  35 ///
  36 /// Template specialization of the standard LLVM dominator tree utility for
  37 /// VPBlockBases.
  38 using VPDominatorTree = DomTreeBase<VPBlockBase>;
  39
  40 using VPDomTreeNode = DomTreeNodeBase<VPBlockBase>;
  41
  42 /// Template specializations of GraphTraits for VPDomTreeNode.
  43 template <>
  44 struct GraphTraits<VPDomTreeNode *>
  45     : public DomTreeGraphTraitsBase<VPDomTreeNode,
  46                                     VPDomTreeNode::const_iterator> {};
  47
  48 template <>
  49 struct GraphTraits<const VPDomTreeNode *>
  50     : public DomTreeGraphTraitsBase<const VPDomTreeNode,
  51                                     VPDomTreeNode::const_iterator> {};
  52 } // namespace llvm
  53 #endif // LLVM_TRANSFORMS_VECTORIZE_VPLANDOMINATORTREE_H