lib/Target/AArch64/AArch64PBQPRegAlloc.h

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