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
24namespace llvm {
25
26template <> 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/// Template specialization of the standard LLVM dominator tree utility for
36/// VPBlockBases.
37class VPDominatorTree : public DominatorTreeBase<VPBlockBase, false> {
38 using Base = DominatorTreeBase<VPBlockBase, false>;
39
40public:
41 VPDominatorTree() = default;
42 explicit VPDominatorTree(VPlan &Plan) { recalculate(Func&: Plan); }
43
44 /// Returns true if \p A properly dominates \p B.
45 using Base::properlyDominates;
46 bool properlyDominates(const VPRecipeBase *A, const VPRecipeBase *B);
47};
48
49using VPDomTreeNode = DomTreeNodeBase<VPBlockBase>;
50
51/// Template specializations of GraphTraits for VPDomTreeNode.
52template <>
53struct GraphTraits<VPDomTreeNode *>
54 : public DomTreeGraphTraitsBase<VPDomTreeNode,
55 VPDomTreeNode::const_iterator> {};
56
57template <>
58struct GraphTraits<const VPDomTreeNode *>
59 : public DomTreeGraphTraitsBase<const VPDomTreeNode,
60 VPDomTreeNode::const_iterator> {};
61} // namespace llvm
62#endif // LLVM_TRANSFORMS_VECTORIZE_VPLANDOMINATORTREE_H
63