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#include "llvm/Support/GenericDomTreeConstruction.h"
24
25namespace llvm {
26
27template <> struct DomTreeNodeTraits<VPBlockBase> {
28 using NodeType = VPBlockBase;
29 using NodePtr = VPBlockBase *;
30 using ParentPtr = VPlan *;
31
32 static NodePtr getEntryNode(ParentPtr Parent) { return Parent->getEntry(); }
33 static ParentPtr getParent(NodePtr B) { return B->getPlan(); }
34};
35
36/// Template specialization of the standard LLVM dominator tree utility for
37/// VPBlockBases.
38class VPDominatorTree : public DominatorTreeBase<VPBlockBase, false> {
39 using Base = DominatorTreeBase<VPBlockBase, false>;
40
41public:
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
49/// Template specialization of the standard LLVM post-dominator tree utility for
50/// VPBlockBases.
51class VPPostDominatorTree : public PostDomTreeBase<VPBlockBase> {
52 using Base = PostDomTreeBase<VPBlockBase>;
53
54public:
55 explicit VPPostDominatorTree(VPlan &Plan) { recalculate(Func&: Plan); }
56};
57
58using VPDomTreeNode = DomTreeNodeBase<VPBlockBase>;
59
60/// Template specializations of GraphTraits for VPDomTreeNode.
61template <>
62struct GraphTraits<VPDomTreeNode *>
63 : public DomTreeGraphTraitsBase<VPDomTreeNode,
64 VPDomTreeNode::const_iterator> {};
65
66template <>
67struct GraphTraits<const VPDomTreeNode *>
68 : public DomTreeGraphTraitsBase<const VPDomTreeNode,
69 VPDomTreeNode::const_iterator> {};
70} // namespace llvm
71#endif // LLVM_TRANSFORMS_VECTORIZE_VPLANDOMINATORTREE_H
72