1 //===- RDFGraph.h -----------------------------------------------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // Target-independent, SSA-based data flow graph for register data flow (RDF) 11 // for a non-SSA program representation (e.g. post-RA machine code). 12 // 13 // 14 // *** Introduction 15 // 16 // The RDF graph is a collection of nodes, each of which denotes some element 17 // of the program. There are two main types of such elements: code and refe- 18 // rences. Conceptually, "code" is something that represents the structure 19 // of the program, e.g. basic block or a statement, while "reference" is an 20 // instance of accessing a register, e.g. a definition or a use. Nodes are 21 // connected with each other based on the structure of the program (such as 22 // blocks, instructions, etc.), and based on the data flow (e.g. reaching 23 // definitions, reached uses, etc.). The single-reaching-definition principle 24 // of SSA is generally observed, although, due to the non-SSA representation 25 // of the program, there are some differences between the graph and a "pure" 26 // SSA representation. 27 // 28 // 29 // *** Implementation remarks 30 // 31 // Since the graph can contain a large number of nodes, memory consumption 32 // was one of the major design considerations. As a result, there is a single 33 // base class NodeBase which defines all members used by all possible derived 34 // classes. The members are arranged in a union, and a derived class cannot 35 // add any data members of its own. Each derived class only defines the 36 // functional interface, i.e. member functions. NodeBase must be a POD, 37 // which implies that all of its members must also be PODs. 38 // Since nodes need to be connected with other nodes, pointers have been 39 // replaced with 32-bit identifiers: each node has an id of type NodeId. 40 // There are mapping functions in the graph that translate between actual 41 // memory addresses and the corresponding identifiers. 42 // A node id of 0 is equivalent to nullptr. 43 // 44 // 45 // *** Structure of the graph 46 // 47 // A code node is always a collection of other nodes. For example, a code 48 // node corresponding to a basic block will contain code nodes corresponding 49 // to instructions. In turn, a code node corresponding to an instruction will 50 // contain a list of reference nodes that correspond to the definitions and 51 // uses of registers in that instruction. The members are arranged into a 52 // circular list, which is yet another consequence of the effort to save 53 // memory: for each member node it should be possible to obtain its owner, 54 // and it should be possible to access all other members. There are other 55 // ways to accomplish that, but the circular list seemed the most natural. 56 // 57 // +- CodeNode -+ 58 // | | <---------------------------------------------------+ 59 // +-+--------+-+ | 60 // |FirstM |LastM | 61 // | +-------------------------------------+ | 62 // | | | 63 // V V | 64 // +----------+ Next +----------+ Next Next +----------+ Next | 65 // | |----->| |-----> ... ----->| |----->-+ 66 // +- Member -+ +- Member -+ +- Member -+ 67 // 68 // The order of members is such that related reference nodes (see below) 69 // should be contiguous on the member list. 70 // 71 // A reference node is a node that encapsulates an access to a register, 72 // in other words, data flowing into or out of a register. There are two 73 // major kinds of reference nodes: defs and uses. A def node will contain 74 // the id of the first reached use, and the id of the first reached def. 75 // Each def and use will contain the id of the reaching def, and also the 76 // id of the next reached def (for def nodes) or use (for use nodes). 77 // The "next node sharing the same reaching def" is denoted as "sibling". 78 // In summary: 79 // - Def node contains: reaching def, sibling, first reached def, and first 80 // reached use. 81 // - Use node contains: reaching def and sibling. 82 // 83 // +-- DefNode --+ 84 // | R2 = ... | <---+--------------------+ 85 // ++---------+--+ | | 86 // |Reached |Reached | | 87 // |Def |Use | | 88 // | | |Reaching |Reaching 89 // | V |Def |Def 90 // | +-- UseNode --+ Sib +-- UseNode --+ Sib Sib 91 // | | ... = R2 |----->| ... = R2 |----> ... ----> 0 92 // | +-------------+ +-------------+ 93 // V 94 // +-- DefNode --+ Sib 95 // | R2 = ... |----> ... 96 // ++---------+--+ 97 // | | 98 // | | 99 // ... ... 100 // 101 // To get a full picture, the circular lists connecting blocks within a 102 // function, instructions within a block, etc. should be superimposed with 103 // the def-def, def-use links shown above. 104 // To illustrate this, consider a small example in a pseudo-assembly: 105 // foo: 106 // add r2, r0, r1 ; r2 = r0+r1 107 // addi r0, r2, 1 ; r0 = r2+1 108 // ret r0 ; return value in r0 109 // 110 // The graph (in a format used by the debugging functions) would look like: 111 // 112 // DFG dump:[ 113 // f1: Function foo 114 // b2: === %bb.0 === preds(0), succs(0): 115 // p3: phi [d4<r0>(,d12,u9):] 116 // p5: phi [d6<r1>(,,u10):] 117 // s7: add [d8<r2>(,,u13):, u9<r0>(d4):, u10<r1>(d6):] 118 // s11: addi [d12<r0>(d4,,u15):, u13<r2>(d8):] 119 // s14: ret [u15<r0>(d12):] 120 // ] 121 // 122 // The f1, b2, p3, etc. are node ids. The letter is prepended to indicate the 123 // kind of the node (i.e. f - function, b - basic block, p - phi, s - state- 124 // ment, d - def, u - use). 125 // The format of a def node is: 126 // dN<R>(rd,d,u):sib, 127 // where 128 // N - numeric node id, 129 // R - register being defined 130 // rd - reaching def, 131 // d - reached def, 132 // u - reached use, 133 // sib - sibling. 134 // The format of a use node is: 135 // uN<R>[!](rd):sib, 136 // where 137 // N - numeric node id, 138 // R - register being used, 139 // rd - reaching def, 140 // sib - sibling. 141 // Possible annotations (usually preceding the node id): 142 // + - preserving def, 143 // ~ - clobbering def, 144 // " - shadow ref (follows the node id), 145 // ! - fixed register (appears after register name). 146 // 147 // The circular lists are not explicit in the dump. 148 // 149 // 150 // *** Node attributes 151 // 152 // NodeBase has a member "Attrs", which is the primary way of determining 153 // the node's characteristics. The fields in this member decide whether 154 // the node is a code node or a reference node (i.e. node's "type"), then 155 // within each type, the "kind" determines what specifically this node 156 // represents. The remaining bits, "flags", contain additional information 157 // that is even more detailed than the "kind". 158 // CodeNode's kinds are: 159 // - Phi: Phi node, members are reference nodes. 160 // - Stmt: Statement, members are reference nodes. 161 // - Block: Basic block, members are instruction nodes (i.e. Phi or Stmt). 162 // - Func: The whole function. The members are basic block nodes. 163 // RefNode's kinds are: 164 // - Use. 165 // - Def. 166 // 167 // Meaning of flags: 168 // - Preserving: applies only to defs. A preserving def is one that can 169 // preserve some of the original bits among those that are included in 170 // the register associated with that def. For example, if R0 is a 32-bit 171 // register, but a def can only change the lower 16 bits, then it will 172 // be marked as preserving. 173 // - Shadow: a reference that has duplicates holding additional reaching 174 // defs (see more below). 175 // - Clobbering: applied only to defs, indicates that the value generated 176 // by this def is unspecified. A typical example would be volatile registers 177 // after function calls. 178 // - Fixed: the register in this def/use cannot be replaced with any other 179 // register. A typical case would be a parameter register to a call, or 180 // the register with the return value from a function. 181 // - Undef: the register in this reference the register is assumed to have 182 // no pre-existing value, even if it appears to be reached by some def. 183 // This is typically used to prevent keeping registers artificially live 184 // in cases when they are defined via predicated instructions. For example: 185 // r0 = add-if-true cond, r10, r11 (1) 186 // r0 = add-if-false cond, r12, r13, implicit r0 (2) 187 // ... = r0 (3) 188 // Before (1), r0 is not intended to be live, and the use of r0 in (3) is 189 // not meant to be reached by any def preceding (1). However, since the 190 // defs in (1) and (2) are both preserving, these properties alone would 191 // imply that the use in (3) may indeed be reached by some prior def. 192 // Adding Undef flag to the def in (1) prevents that. The Undef flag 193 // may be applied to both defs and uses. 194 // - Dead: applies only to defs. The value coming out of a "dead" def is 195 // assumed to be unused, even if the def appears to be reaching other defs 196 // or uses. The motivation for this flag comes from dead defs on function 197 // calls: there is no way to determine if such a def is dead without 198 // analyzing the target's ABI. Hence the graph should contain this info, 199 // as it is unavailable otherwise. On the other hand, a def without any 200 // uses on a typical instruction is not the intended target for this flag. 201 // 202 // *** Shadow references 203 // 204 // It may happen that a super-register can have two (or more) non-overlapping 205 // sub-registers. When both of these sub-registers are defined and followed 206 // by a use of the super-register, the use of the super-register will not 207 // have a unique reaching def: both defs of the sub-registers need to be 208 // accounted for. In such cases, a duplicate use of the super-register is 209 // added and it points to the extra reaching def. Both uses are marked with 210 // a flag "shadow". Example: 211 // Assume t0 is a super-register of r0 and r1, r0 and r1 do not overlap: 212 // set r0, 1 ; r0 = 1 213 // set r1, 1 ; r1 = 1 214 // addi t1, t0, 1 ; t1 = t0+1 215 // 216 // The DFG: 217 // s1: set [d2<r0>(,,u9):] 218 // s3: set [d4<r1>(,,u10):] 219 // s5: addi [d6<t1>(,,):, u7"<t0>(d2):, u8"<t0>(d4):] 220 // 221 // The statement s5 has two use nodes for t0: u7" and u9". The quotation 222 // mark " indicates that the node is a shadow. 223 // 224 225 #ifndef LLVM_LIB_TARGET_HEXAGON_RDFGRAPH_H 226 #define LLVM_LIB_TARGET_HEXAGON_RDFGRAPH_H 227 228 #include "RDFRegisters.h" 229 #include "llvm/ADT/SmallVector.h" 230 #include "llvm/MC/LaneBitmask.h" 231 #include "llvm/Support/Allocator.h" 232 #include "llvm/Support/MathExtras.h" 233 #include <cassert> 234 #include <cstdint> 235 #include <cstring> 236 #include <map> 237 #include <set> 238 #include <unordered_map> 239 #include <utility> 240 #include <vector> 241 242 // RDF uses uint32_t to refer to registers. This is to ensure that the type 243 // size remains specific. In other places, registers are often stored using 244 // unsigned. 245 static_assert(sizeof(uint32_t) == sizeof(unsigned), "Those should be equal"); 246 247 namespace llvm { 248 249 class MachineBasicBlock; 250 class MachineDominanceFrontier; 251 class MachineDominatorTree; 252 class MachineFunction; 253 class MachineInstr; 254 class MachineOperand; 255 class raw_ostream; 256 class TargetInstrInfo; 257 class TargetRegisterInfo; 258 259 namespace rdf { 260 261 using NodeId = uint32_t; 262 263 struct DataFlowGraph; 264 265 struct NodeAttrs { 266 enum : uint16_t { 267 None = 0x0000, // Nothing 268 269 // Types: 2 bits 270 TypeMask = 0x0003, 271 Code = 0x0001, // 01, Container 272 Ref = 0x0002, // 10, Reference 273 274 // Kind: 3 bits 275 KindMask = 0x0007 << 2, 276 Def = 0x0001 << 2, // 001 277 Use = 0x0002 << 2, // 010 278 Phi = 0x0003 << 2, // 011 279 Stmt = 0x0004 << 2, // 100 280 Block = 0x0005 << 2, // 101 281 Func = 0x0006 << 2, // 110 282 283 // Flags: 7 bits for now 284 FlagMask = 0x007F << 5, 285 Shadow = 0x0001 << 5, // 0000001, Has extra reaching defs. 286 Clobbering = 0x0002 << 5, // 0000010, Produces unspecified values. 287 PhiRef = 0x0004 << 5, // 0000100, Member of PhiNode. 288 Preserving = 0x0008 << 5, // 0001000, Def can keep original bits. 289 Fixed = 0x0010 << 5, // 0010000, Fixed register. 290 Undef = 0x0020 << 5, // 0100000, Has no pre-existing value. 291 Dead = 0x0040 << 5, // 1000000, Does not define a value. 292 }; 293 typeNodeAttrs294 static uint16_t type(uint16_t T) { return T & TypeMask; } kindNodeAttrs295 static uint16_t kind(uint16_t T) { return T & KindMask; } flagsNodeAttrs296 static uint16_t flags(uint16_t T) { return T & FlagMask; } 297 set_typeNodeAttrs298 static uint16_t set_type(uint16_t A, uint16_t T) { 299 return (A & ~TypeMask) | T; 300 } 301 set_kindNodeAttrs302 static uint16_t set_kind(uint16_t A, uint16_t K) { 303 return (A & ~KindMask) | K; 304 } 305 set_flagsNodeAttrs306 static uint16_t set_flags(uint16_t A, uint16_t F) { 307 return (A & ~FlagMask) | F; 308 } 309 310 // Test if A contains B. containsNodeAttrs311 static bool contains(uint16_t A, uint16_t B) { 312 if (type(A) != Code) 313 return false; 314 uint16_t KB = kind(B); 315 switch (kind(A)) { 316 case Func: 317 return KB == Block; 318 case Block: 319 return KB == Phi || KB == Stmt; 320 case Phi: 321 case Stmt: 322 return type(B) == Ref; 323 } 324 return false; 325 } 326 }; 327 328 struct BuildOptions { 329 enum : unsigned { 330 None = 0x00, 331 KeepDeadPhis = 0x01, // Do not remove dead phis during build. 332 }; 333 }; 334 335 template <typename T> struct NodeAddr { 336 NodeAddr() = default; NodeAddrNodeAddr337 NodeAddr(T A, NodeId I) : Addr(A), Id(I) {} 338 339 // Type cast (casting constructor). The reason for having this class 340 // instead of std::pair. NodeAddrNodeAddr341 template <typename S> NodeAddr(const NodeAddr<S> &NA) 342 : Addr(static_cast<T>(NA.Addr)), Id(NA.Id) {} 343 344 bool operator== (const NodeAddr<T> &NA) const { 345 assert((Addr == NA.Addr) == (Id == NA.Id)); 346 return Addr == NA.Addr; 347 } 348 bool operator!= (const NodeAddr<T> &NA) const { 349 return !operator==(NA); 350 } 351 352 T Addr = nullptr; 353 NodeId Id = 0; 354 }; 355 356 struct NodeBase; 357 358 // Fast memory allocation and translation between node id and node address. 359 // This is really the same idea as the one underlying the "bump pointer 360 // allocator", the difference being in the translation. A node id is 361 // composed of two components: the index of the block in which it was 362 // allocated, and the index within the block. With the default settings, 363 // where the number of nodes per block is 4096, the node id (minus 1) is: 364 // 365 // bit position: 11 0 366 // +----------------------------+--------------+ 367 // | Index of the block |Index in block| 368 // +----------------------------+--------------+ 369 // 370 // The actual node id is the above plus 1, to avoid creating a node id of 0. 371 // 372 // This method significantly improved the build time, compared to using maps 373 // (std::unordered_map or DenseMap) to translate between pointers and ids. 374 struct NodeAllocator { 375 // Amount of storage for a single node. 376 enum { NodeMemSize = 32 }; 377 378 NodeAllocator(uint32_t NPB = 4096) NodesPerBlockNodeAllocator379 : NodesPerBlock(NPB), BitsPerIndex(Log2_32(NPB)), 380 IndexMask((1 << BitsPerIndex)-1) { 381 assert(isPowerOf2_32(NPB)); 382 } 383 ptrNodeAllocator384 NodeBase *ptr(NodeId N) const { 385 uint32_t N1 = N-1; 386 uint32_t BlockN = N1 >> BitsPerIndex; 387 uint32_t Offset = (N1 & IndexMask) * NodeMemSize; 388 return reinterpret_cast<NodeBase*>(Blocks[BlockN]+Offset); 389 } 390 391 NodeId id(const NodeBase *P) const; 392 NodeAddr<NodeBase*> New(); 393 void clear(); 394 395 private: 396 void startNewBlock(); 397 bool needNewBlock(); 398 makeIdNodeAllocator399 uint32_t makeId(uint32_t Block, uint32_t Index) const { 400 // Add 1 to the id, to avoid the id of 0, which is treated as "null". 401 return ((Block << BitsPerIndex) | Index) + 1; 402 } 403 404 const uint32_t NodesPerBlock; 405 const uint32_t BitsPerIndex; 406 const uint32_t IndexMask; 407 char *ActiveEnd = nullptr; 408 std::vector<char*> Blocks; 409 using AllocatorTy = BumpPtrAllocatorImpl<MallocAllocator, 65536>; 410 AllocatorTy MemPool; 411 }; 412 413 using RegisterSet = std::set<RegisterRef>; 414 415 struct TargetOperandInfo { TargetOperandInfoTargetOperandInfo416 TargetOperandInfo(const TargetInstrInfo &tii) : TII(tii) {} 417 virtual ~TargetOperandInfo() = default; 418 419 virtual bool isPreserving(const MachineInstr &In, unsigned OpNum) const; 420 virtual bool isClobbering(const MachineInstr &In, unsigned OpNum) const; 421 virtual bool isFixedReg(const MachineInstr &In, unsigned OpNum) const; 422 423 const TargetInstrInfo &TII; 424 }; 425 426 // Packed register reference. Only used for storage. 427 struct PackedRegisterRef { 428 RegisterId Reg; 429 uint32_t MaskId; 430 }; 431 432 struct LaneMaskIndex : private IndexedSet<LaneBitmask> { 433 LaneMaskIndex() = default; 434 getLaneMaskForIndexLaneMaskIndex435 LaneBitmask getLaneMaskForIndex(uint32_t K) const { 436 return K == 0 ? LaneBitmask::getAll() : get(K); 437 } 438 getIndexForLaneMaskLaneMaskIndex439 uint32_t getIndexForLaneMask(LaneBitmask LM) { 440 assert(LM.any()); 441 return LM.all() ? 0 : insert(LM); 442 } 443 getIndexForLaneMaskLaneMaskIndex444 uint32_t getIndexForLaneMask(LaneBitmask LM) const { 445 assert(LM.any()); 446 return LM.all() ? 0 : find(LM); 447 } 448 }; 449 450 struct NodeBase { 451 public: 452 // Make sure this is a POD. 453 NodeBase() = default; 454 getTypeNodeBase455 uint16_t getType() const { return NodeAttrs::type(Attrs); } getKindNodeBase456 uint16_t getKind() const { return NodeAttrs::kind(Attrs); } getFlagsNodeBase457 uint16_t getFlags() const { return NodeAttrs::flags(Attrs); } getNextNodeBase458 NodeId getNext() const { return Next; } 459 getAttrsNodeBase460 uint16_t getAttrs() const { return Attrs; } setAttrsNodeBase461 void setAttrs(uint16_t A) { Attrs = A; } setFlagsNodeBase462 void setFlags(uint16_t F) { setAttrs(NodeAttrs::set_flags(getAttrs(), F)); } 463 464 // Insert node NA after "this" in the circular chain. 465 void append(NodeAddr<NodeBase*> NA); 466 467 // Initialize all members to 0. initNodeBase468 void init() { memset(this, 0, sizeof *this); } 469 setNextNodeBase470 void setNext(NodeId N) { Next = N; } 471 472 protected: 473 uint16_t Attrs; 474 uint16_t Reserved; 475 NodeId Next; // Id of the next node in the circular chain. 476 // Definitions of nested types. Using anonymous nested structs would make 477 // this class definition clearer, but unnamed structs are not a part of 478 // the standard. 479 struct Def_struct { 480 NodeId DD, DU; // Ids of the first reached def and use. 481 }; 482 struct PhiU_struct { 483 NodeId PredB; // Id of the predecessor block for a phi use. 484 }; 485 struct Code_struct { 486 void *CP; // Pointer to the actual code. 487 NodeId FirstM, LastM; // Id of the first member and last. 488 }; 489 struct Ref_struct { 490 NodeId RD, Sib; // Ids of the reaching def and the sibling. 491 union { 492 Def_struct Def; 493 PhiU_struct PhiU; 494 }; 495 union { 496 MachineOperand *Op; // Non-phi refs point to a machine operand. 497 PackedRegisterRef PR; // Phi refs store register info directly. 498 }; 499 }; 500 501 // The actual payload. 502 union { 503 Ref_struct Ref; 504 Code_struct Code; 505 }; 506 }; 507 // The allocator allocates chunks of 32 bytes for each node. The fact that 508 // each node takes 32 bytes in memory is used for fast translation between 509 // the node id and the node address. 510 static_assert(sizeof(NodeBase) <= NodeAllocator::NodeMemSize, 511 "NodeBase must be at most NodeAllocator::NodeMemSize bytes"); 512 513 using NodeList = SmallVector<NodeAddr<NodeBase *>, 4>; 514 using NodeSet = std::set<NodeId>; 515 516 struct RefNode : public NodeBase { 517 RefNode() = default; 518 519 RegisterRef getRegRef(const DataFlowGraph &G) const; 520 getOpRefNode521 MachineOperand &getOp() { 522 assert(!(getFlags() & NodeAttrs::PhiRef)); 523 return *Ref.Op; 524 } 525 526 void setRegRef(RegisterRef RR, DataFlowGraph &G); 527 void setRegRef(MachineOperand *Op, DataFlowGraph &G); 528 getReachingDefRefNode529 NodeId getReachingDef() const { 530 return Ref.RD; 531 } setReachingDefRefNode532 void setReachingDef(NodeId RD) { 533 Ref.RD = RD; 534 } 535 getSiblingRefNode536 NodeId getSibling() const { 537 return Ref.Sib; 538 } setSiblingRefNode539 void setSibling(NodeId Sib) { 540 Ref.Sib = Sib; 541 } 542 isUseRefNode543 bool isUse() const { 544 assert(getType() == NodeAttrs::Ref); 545 return getKind() == NodeAttrs::Use; 546 } 547 isDefRefNode548 bool isDef() const { 549 assert(getType() == NodeAttrs::Ref); 550 return getKind() == NodeAttrs::Def; 551 } 552 553 template <typename Predicate> 554 NodeAddr<RefNode*> getNextRef(RegisterRef RR, Predicate P, bool NextOnly, 555 const DataFlowGraph &G); 556 NodeAddr<NodeBase*> getOwner(const DataFlowGraph &G); 557 }; 558 559 struct DefNode : public RefNode { getReachedDefDefNode560 NodeId getReachedDef() const { 561 return Ref.Def.DD; 562 } setReachedDefDefNode563 void setReachedDef(NodeId D) { 564 Ref.Def.DD = D; 565 } getReachedUseDefNode566 NodeId getReachedUse() const { 567 return Ref.Def.DU; 568 } setReachedUseDefNode569 void setReachedUse(NodeId U) { 570 Ref.Def.DU = U; 571 } 572 573 void linkToDef(NodeId Self, NodeAddr<DefNode*> DA); 574 }; 575 576 struct UseNode : public RefNode { 577 void linkToDef(NodeId Self, NodeAddr<DefNode*> DA); 578 }; 579 580 struct PhiUseNode : public UseNode { getPredecessorPhiUseNode581 NodeId getPredecessor() const { 582 assert(getFlags() & NodeAttrs::PhiRef); 583 return Ref.PhiU.PredB; 584 } setPredecessorPhiUseNode585 void setPredecessor(NodeId B) { 586 assert(getFlags() & NodeAttrs::PhiRef); 587 Ref.PhiU.PredB = B; 588 } 589 }; 590 591 struct CodeNode : public NodeBase { getCodeCodeNode592 template <typename T> T getCode() const { 593 return static_cast<T>(Code.CP); 594 } setCodeCodeNode595 void setCode(void *C) { 596 Code.CP = C; 597 } 598 599 NodeAddr<NodeBase*> getFirstMember(const DataFlowGraph &G) const; 600 NodeAddr<NodeBase*> getLastMember(const DataFlowGraph &G) const; 601 void addMember(NodeAddr<NodeBase*> NA, const DataFlowGraph &G); 602 void addMemberAfter(NodeAddr<NodeBase*> MA, NodeAddr<NodeBase*> NA, 603 const DataFlowGraph &G); 604 void removeMember(NodeAddr<NodeBase*> NA, const DataFlowGraph &G); 605 606 NodeList members(const DataFlowGraph &G) const; 607 template <typename Predicate> 608 NodeList members_if(Predicate P, const DataFlowGraph &G) const; 609 }; 610 611 struct InstrNode : public CodeNode { 612 NodeAddr<NodeBase*> getOwner(const DataFlowGraph &G); 613 }; 614 615 struct PhiNode : public InstrNode { getCodePhiNode616 MachineInstr *getCode() const { 617 return nullptr; 618 } 619 }; 620 621 struct StmtNode : public InstrNode { getCodeStmtNode622 MachineInstr *getCode() const { 623 return CodeNode::getCode<MachineInstr*>(); 624 } 625 }; 626 627 struct BlockNode : public CodeNode { getCodeBlockNode628 MachineBasicBlock *getCode() const { 629 return CodeNode::getCode<MachineBasicBlock*>(); 630 } 631 632 void addPhi(NodeAddr<PhiNode*> PA, const DataFlowGraph &G); 633 }; 634 635 struct FuncNode : public CodeNode { getCodeFuncNode636 MachineFunction *getCode() const { 637 return CodeNode::getCode<MachineFunction*>(); 638 } 639 640 NodeAddr<BlockNode*> findBlock(const MachineBasicBlock *BB, 641 const DataFlowGraph &G) const; 642 NodeAddr<BlockNode*> getEntryBlock(const DataFlowGraph &G); 643 }; 644 645 struct DataFlowGraph { 646 DataFlowGraph(MachineFunction &mf, const TargetInstrInfo &tii, 647 const TargetRegisterInfo &tri, const MachineDominatorTree &mdt, 648 const MachineDominanceFrontier &mdf, const TargetOperandInfo &toi); 649 650 NodeBase *ptr(NodeId N) const; ptrDataFlowGraph651 template <typename T> T ptr(NodeId N) const { 652 return static_cast<T>(ptr(N)); 653 } 654 655 NodeId id(const NodeBase *P) const; 656 addrDataFlowGraph657 template <typename T> NodeAddr<T> addr(NodeId N) const { 658 return { ptr<T>(N), N }; 659 } 660 getFuncDataFlowGraph661 NodeAddr<FuncNode*> getFunc() const { return Func; } getMFDataFlowGraph662 MachineFunction &getMF() const { return MF; } getTIIDataFlowGraph663 const TargetInstrInfo &getTII() const { return TII; } getTRIDataFlowGraph664 const TargetRegisterInfo &getTRI() const { return TRI; } getPRIDataFlowGraph665 const PhysicalRegisterInfo &getPRI() const { return PRI; } getDTDataFlowGraph666 const MachineDominatorTree &getDT() const { return MDT; } getDFDataFlowGraph667 const MachineDominanceFrontier &getDF() const { return MDF; } getLiveInsDataFlowGraph668 const RegisterAggr &getLiveIns() const { return LiveIns; } 669 670 struct DefStack { 671 DefStack() = default; 672 emptyDataFlowGraph::DefStack673 bool empty() const { return Stack.empty() || top() == bottom(); } 674 675 private: 676 using value_type = NodeAddr<DefNode *>; 677 struct Iterator { 678 using value_type = DefStack::value_type; 679 upDataFlowGraph::DefStack::Iterator680 Iterator &up() { Pos = DS.nextUp(Pos); return *this; } downDataFlowGraph::DefStack::Iterator681 Iterator &down() { Pos = DS.nextDown(Pos); return *this; } 682 683 value_type operator*() const { 684 assert(Pos >= 1); 685 return DS.Stack[Pos-1]; 686 } 687 const value_type *operator->() const { 688 assert(Pos >= 1); 689 return &DS.Stack[Pos-1]; 690 } 691 bool operator==(const Iterator &It) const { return Pos == It.Pos; } 692 bool operator!=(const Iterator &It) const { return Pos != It.Pos; } 693 694 private: 695 friend struct DefStack; 696 697 Iterator(const DefStack &S, bool Top); 698 699 // Pos-1 is the index in the StorageType object that corresponds to 700 // the top of the DefStack. 701 const DefStack &DS; 702 unsigned Pos; 703 }; 704 705 public: 706 using iterator = Iterator; 707 topDataFlowGraph::DefStack708 iterator top() const { return Iterator(*this, true); } bottomDataFlowGraph::DefStack709 iterator bottom() const { return Iterator(*this, false); } 710 unsigned size() const; 711 pushDataFlowGraph::DefStack712 void push(NodeAddr<DefNode*> DA) { Stack.push_back(DA); } 713 void pop(); 714 void start_block(NodeId N); 715 void clear_block(NodeId N); 716 717 private: 718 friend struct Iterator; 719 720 using StorageType = std::vector<value_type>; 721 722 bool isDelimiter(const StorageType::value_type &P, NodeId N = 0) const { 723 return (P.Addr == nullptr) && (N == 0 || P.Id == N); 724 } 725 726 unsigned nextUp(unsigned P) const; 727 unsigned nextDown(unsigned P) const; 728 729 StorageType Stack; 730 }; 731 732 // Make this std::unordered_map for speed of accessing elements. 733 // Map: Register (physical or virtual) -> DefStack 734 using DefStackMap = std::unordered_map<RegisterId, DefStack>; 735 736 void build(unsigned Options = BuildOptions::None); 737 void pushAllDefs(NodeAddr<InstrNode*> IA, DefStackMap &DM); 738 void markBlock(NodeId B, DefStackMap &DefM); 739 void releaseBlock(NodeId B, DefStackMap &DefM); 740 packDataFlowGraph741 PackedRegisterRef pack(RegisterRef RR) { 742 return { RR.Reg, LMI.getIndexForLaneMask(RR.Mask) }; 743 } packDataFlowGraph744 PackedRegisterRef pack(RegisterRef RR) const { 745 return { RR.Reg, LMI.getIndexForLaneMask(RR.Mask) }; 746 } unpackDataFlowGraph747 RegisterRef unpack(PackedRegisterRef PR) const { 748 return RegisterRef(PR.Reg, LMI.getLaneMaskForIndex(PR.MaskId)); 749 } 750 751 RegisterRef makeRegRef(unsigned Reg, unsigned Sub) const; 752 RegisterRef makeRegRef(const MachineOperand &Op) const; 753 RegisterRef restrictRef(RegisterRef AR, RegisterRef BR) const; 754 755 NodeAddr<RefNode*> getNextRelated(NodeAddr<InstrNode*> IA, 756 NodeAddr<RefNode*> RA) const; 757 NodeAddr<RefNode*> getNextImp(NodeAddr<InstrNode*> IA, 758 NodeAddr<RefNode*> RA, bool Create); 759 NodeAddr<RefNode*> getNextImp(NodeAddr<InstrNode*> IA, 760 NodeAddr<RefNode*> RA) const; 761 NodeAddr<RefNode*> getNextShadow(NodeAddr<InstrNode*> IA, 762 NodeAddr<RefNode*> RA, bool Create); 763 NodeAddr<RefNode*> getNextShadow(NodeAddr<InstrNode*> IA, 764 NodeAddr<RefNode*> RA) const; 765 766 NodeList getRelatedRefs(NodeAddr<InstrNode*> IA, 767 NodeAddr<RefNode*> RA) const; 768 findBlockDataFlowGraph769 NodeAddr<BlockNode*> findBlock(MachineBasicBlock *BB) const { 770 return BlockNodes.at(BB); 771 } 772 unlinkUseDataFlowGraph773 void unlinkUse(NodeAddr<UseNode*> UA, bool RemoveFromOwner) { 774 unlinkUseDF(UA); 775 if (RemoveFromOwner) 776 removeFromOwner(UA); 777 } 778 unlinkDefDataFlowGraph779 void unlinkDef(NodeAddr<DefNode*> DA, bool RemoveFromOwner) { 780 unlinkDefDF(DA); 781 if (RemoveFromOwner) 782 removeFromOwner(DA); 783 } 784 785 // Some useful filters. 786 template <uint16_t Kind> IsRefDataFlowGraph787 static bool IsRef(const NodeAddr<NodeBase*> BA) { 788 return BA.Addr->getType() == NodeAttrs::Ref && 789 BA.Addr->getKind() == Kind; 790 } 791 792 template <uint16_t Kind> IsCodeDataFlowGraph793 static bool IsCode(const NodeAddr<NodeBase*> BA) { 794 return BA.Addr->getType() == NodeAttrs::Code && 795 BA.Addr->getKind() == Kind; 796 } 797 IsDefDataFlowGraph798 static bool IsDef(const NodeAddr<NodeBase*> BA) { 799 return BA.Addr->getType() == NodeAttrs::Ref && 800 BA.Addr->getKind() == NodeAttrs::Def; 801 } 802 IsUseDataFlowGraph803 static bool IsUse(const NodeAddr<NodeBase*> BA) { 804 return BA.Addr->getType() == NodeAttrs::Ref && 805 BA.Addr->getKind() == NodeAttrs::Use; 806 } 807 IsPhiDataFlowGraph808 static bool IsPhi(const NodeAddr<NodeBase*> BA) { 809 return BA.Addr->getType() == NodeAttrs::Code && 810 BA.Addr->getKind() == NodeAttrs::Phi; 811 } 812 IsPreservingDefDataFlowGraph813 static bool IsPreservingDef(const NodeAddr<DefNode*> DA) { 814 uint16_t Flags = DA.Addr->getFlags(); 815 return (Flags & NodeAttrs::Preserving) && !(Flags & NodeAttrs::Undef); 816 } 817 818 private: 819 void reset(); 820 821 RegisterSet getLandingPadLiveIns() const; 822 823 NodeAddr<NodeBase*> newNode(uint16_t Attrs); 824 NodeAddr<NodeBase*> cloneNode(const NodeAddr<NodeBase*> B); 825 NodeAddr<UseNode*> newUse(NodeAddr<InstrNode*> Owner, 826 MachineOperand &Op, uint16_t Flags = NodeAttrs::None); 827 NodeAddr<PhiUseNode*> newPhiUse(NodeAddr<PhiNode*> Owner, 828 RegisterRef RR, NodeAddr<BlockNode*> PredB, 829 uint16_t Flags = NodeAttrs::PhiRef); 830 NodeAddr<DefNode*> newDef(NodeAddr<InstrNode*> Owner, 831 MachineOperand &Op, uint16_t Flags = NodeAttrs::None); 832 NodeAddr<DefNode*> newDef(NodeAddr<InstrNode*> Owner, 833 RegisterRef RR, uint16_t Flags = NodeAttrs::PhiRef); 834 NodeAddr<PhiNode*> newPhi(NodeAddr<BlockNode*> Owner); 835 NodeAddr<StmtNode*> newStmt(NodeAddr<BlockNode*> Owner, 836 MachineInstr *MI); 837 NodeAddr<BlockNode*> newBlock(NodeAddr<FuncNode*> Owner, 838 MachineBasicBlock *BB); 839 NodeAddr<FuncNode*> newFunc(MachineFunction *MF); 840 841 template <typename Predicate> 842 std::pair<NodeAddr<RefNode*>,NodeAddr<RefNode*>> 843 locateNextRef(NodeAddr<InstrNode*> IA, NodeAddr<RefNode*> RA, 844 Predicate P) const; 845 846 using BlockRefsMap = std::map<NodeId, RegisterSet>; 847 848 void buildStmt(NodeAddr<BlockNode*> BA, MachineInstr &In); 849 void recordDefsForDF(BlockRefsMap &PhiM, NodeAddr<BlockNode*> BA); 850 void buildPhis(BlockRefsMap &PhiM, RegisterSet &AllRefs, 851 NodeAddr<BlockNode*> BA); 852 void removeUnusedPhis(); 853 854 void pushClobbers(NodeAddr<InstrNode*> IA, DefStackMap &DM); 855 void pushDefs(NodeAddr<InstrNode*> IA, DefStackMap &DM); 856 template <typename T> void linkRefUp(NodeAddr<InstrNode*> IA, 857 NodeAddr<T> TA, DefStack &DS); 858 template <typename Predicate> void linkStmtRefs(DefStackMap &DefM, 859 NodeAddr<StmtNode*> SA, Predicate P); 860 void linkBlockRefs(DefStackMap &DefM, NodeAddr<BlockNode*> BA); 861 862 void unlinkUseDF(NodeAddr<UseNode*> UA); 863 void unlinkDefDF(NodeAddr<DefNode*> DA); 864 removeFromOwnerDataFlowGraph865 void removeFromOwner(NodeAddr<RefNode*> RA) { 866 NodeAddr<InstrNode*> IA = RA.Addr->getOwner(*this); 867 IA.Addr->removeMember(RA, *this); 868 } 869 870 MachineFunction &MF; 871 const TargetInstrInfo &TII; 872 const TargetRegisterInfo &TRI; 873 const PhysicalRegisterInfo PRI; 874 const MachineDominatorTree &MDT; 875 const MachineDominanceFrontier &MDF; 876 const TargetOperandInfo &TOI; 877 878 RegisterAggr LiveIns; 879 NodeAddr<FuncNode*> Func; 880 NodeAllocator Memory; 881 // Local map: MachineBasicBlock -> NodeAddr<BlockNode*> 882 std::map<MachineBasicBlock*,NodeAddr<BlockNode*>> BlockNodes; 883 // Lane mask map. 884 LaneMaskIndex LMI; 885 }; // struct DataFlowGraph 886 887 template <typename Predicate> getNextRef(RegisterRef RR,Predicate P,bool NextOnly,const DataFlowGraph & G)888 NodeAddr<RefNode*> RefNode::getNextRef(RegisterRef RR, Predicate P, 889 bool NextOnly, const DataFlowGraph &G) { 890 // Get the "Next" reference in the circular list that references RR and 891 // satisfies predicate "Pred". 892 auto NA = G.addr<NodeBase*>(getNext()); 893 894 while (NA.Addr != this) { 895 if (NA.Addr->getType() == NodeAttrs::Ref) { 896 NodeAddr<RefNode*> RA = NA; 897 if (RA.Addr->getRegRef(G) == RR && P(NA)) 898 return NA; 899 if (NextOnly) 900 break; 901 NA = G.addr<NodeBase*>(NA.Addr->getNext()); 902 } else { 903 // We've hit the beginning of the chain. 904 assert(NA.Addr->getType() == NodeAttrs::Code); 905 NodeAddr<CodeNode*> CA = NA; 906 NA = CA.Addr->getFirstMember(G); 907 } 908 } 909 // Return the equivalent of "nullptr" if such a node was not found. 910 return NodeAddr<RefNode*>(); 911 } 912 913 template <typename Predicate> members_if(Predicate P,const DataFlowGraph & G)914 NodeList CodeNode::members_if(Predicate P, const DataFlowGraph &G) const { 915 NodeList MM; 916 auto M = getFirstMember(G); 917 if (M.Id == 0) 918 return MM; 919 920 while (M.Addr != this) { 921 if (P(M)) 922 MM.push_back(M); 923 M = G.addr<NodeBase*>(M.Addr->getNext()); 924 } 925 return MM; 926 } 927 928 template <typename T> struct Print; 929 template <typename T> 930 raw_ostream &operator<< (raw_ostream &OS, const Print<T> &P); 931 932 template <typename T> 933 struct Print { PrintPrint934 Print(const T &x, const DataFlowGraph &g) : Obj(x), G(g) {} 935 936 const T &Obj; 937 const DataFlowGraph &G; 938 }; 939 940 template <typename T> 941 struct PrintNode : Print<NodeAddr<T>> { PrintNodePrintNode942 PrintNode(const NodeAddr<T> &x, const DataFlowGraph &g) 943 : Print<NodeAddr<T>>(x, g) {} 944 }; 945 946 } // end namespace rdf 947 948 } // end namespace llvm 949 950 #endif // LLVM_LIB_TARGET_HEXAGON_RDFGRAPH_H 951