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
41