| 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 |
| 54 |