lib/Target/AArch64/AArch64PBQPRegAlloc.h

   1 //==- AArch64PBQPRegAlloc.h - AArch64 specific PBQP constraints --*- 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 #ifndef LLVM_LIB_TARGET_AARCH64_AARCH64PBQPREGALOC_H
  11 #define LLVM_LIB_TARGET_AARCH64_AARCH64PBQPREGALOC_H
  12
  13 #include "llvm/ADT/SetVector.h"
  14 #include "llvm/CodeGen/PBQPRAConstraint.h"
  15
  16 namespace llvm {
  17
  18 class TargetRegisterInfo;
  19
  20 /// Add the accumulator chaining constraint to a PBQP graph
  21 class A57ChainingConstraint : public PBQPRAConstraint {
  22 public:
  23   // Add A57 specific constraints to the PBQP graph.
  24   void apply(PBQPRAGraph &G) override;
  25
  26 private:
  27   SmallSetVector<unsigned, 32> Chains;
  28   const TargetRegisterInfo *TRI;
  29
  30   // Add the accumulator chaining constraint, inside the chain, i.e. so that
  31   // parity(Rd) == parity(Ra).
  32   // \return true if a constraint was added
  33   bool addIntraChainConstraint(PBQPRAGraph &G, unsigned Rd, unsigned Ra);
  34
  35   // Add constraints between existing chains
  36   void addInterChainConstraint(PBQPRAGraph &G, unsigned Rd, unsigned Ra);
  37 };
  38
  39 } // end namespace llvm
  40
  41 #endif // LLVM_LIB_TARGET_AARCH64_AARCH64PBQPREGALOC_H