14 #ifndef LLVM_CODEGEN_PBQP_REDUCTIONRULES_H
15 #define LLVM_CODEGEN_PBQP_REDUCTIONRULES_H
28 template <
typename GraphT>
29 void applyR1(GraphT &
G,
typename GraphT::NodeId NId) {
30 typedef typename GraphT::NodeId NodeId;
31 typedef typename GraphT::EdgeId EdgeId;
32 typedef typename GraphT::Vector
Vector;
34 typedef typename GraphT::RawVector RawVector;
36 assert(G.getNodeDegree(NId) == 1 &&
37 "R1 applied to node with degree != 1.");
39 EdgeId EId = *G.adjEdgeIds(NId).begin();
40 NodeId MId = G.getEdgeOtherNodeId(EId, NId);
42 const Matrix &ECosts = G.getEdgeCosts(EId);
43 const Vector &XCosts = G.getNodeCosts(NId);
44 RawVector YCosts = G.getNodeCosts(MId);
47 if (NId == G.getEdgeNode1Id(EId)) {
48 for (
unsigned j = 0; j < YCosts.getLength(); ++j) {
49 PBQPNum Min = ECosts[0][j] + XCosts[0];
50 for (
unsigned i = 1; i < XCosts.getLength(); ++i) {
51 PBQPNum C = ECosts[i][j] + XCosts[i];
58 for (
unsigned i = 0; i < YCosts.getLength(); ++i) {
59 PBQPNum Min = ECosts[i][0] + XCosts[0];
60 for (
unsigned j = 1; j < XCosts.getLength(); ++j) {
61 PBQPNum C = ECosts[i][j] + XCosts[j];
68 G.setNodeCosts(MId, YCosts);
69 G.disconnectEdge(EId, MId);
72 template <
typename GraphT>
73 void applyR2(GraphT &
G,
typename GraphT::NodeId NId) {
74 typedef typename GraphT::NodeId NodeId;
75 typedef typename GraphT::EdgeId EdgeId;
76 typedef typename GraphT::Vector
Vector;
78 typedef typename GraphT::RawMatrix RawMatrix;
80 assert(G.getNodeDegree(NId) == 2 &&
81 "R2 applied to node with degree != 2.");
83 const Vector &XCosts = G.getNodeCosts(NId);
85 typename GraphT::AdjEdgeItr AEItr = G.adjEdgeIds(NId).begin();
86 EdgeId YXEId = *AEItr,
89 NodeId YNId = G.getEdgeOtherNodeId(YXEId, NId),
90 ZNId = G.getEdgeOtherNodeId(ZXEId, NId);
92 bool FlipEdge1 = (G.getEdgeNode1Id(YXEId) == NId),
93 FlipEdge2 = (G.getEdgeNode1Id(ZXEId) == NId);
95 const Matrix *YXECosts = FlipEdge1 ?
96 new Matrix(G.getEdgeCosts(YXEId).transpose()) :
97 &G.getEdgeCosts(YXEId);
99 const Matrix *ZXECosts = FlipEdge2 ?
100 new Matrix(G.getEdgeCosts(ZXEId).transpose()) :
101 &G.getEdgeCosts(ZXEId);
103 unsigned XLen = XCosts.getLength(),
107 RawMatrix Delta(YLen, ZLen);
109 for (
unsigned i = 0; i < YLen; ++i) {
110 for (
unsigned j = 0; j < ZLen; ++j) {
111 PBQPNum Min = (*YXECosts)[i][0] + (*ZXECosts)[j][0] + XCosts[0];
112 for (
unsigned k = 1; k < XLen; ++k) {
113 PBQPNum C = (*YXECosts)[i][k] + (*ZXECosts)[j][k] + XCosts[k];
128 EdgeId YZEId = G.findEdge(YNId, ZNId);
130 if (YZEId == G.invalidEdgeId()) {
131 YZEId = G.addEdge(YNId, ZNId, Delta);
133 const Matrix &YZECosts = G.getEdgeCosts(YZEId);
134 if (YNId == G.getEdgeNode1Id(YZEId)) {
135 G.updateEdgeCosts(YZEId, Delta + YZECosts);
137 G.updateEdgeCosts(YZEId, Delta.transpose() + YZECosts);
141 G.disconnectEdge(YXEId, YNId);
142 G.disconnectEdge(ZXEId, ZNId);
149 template <
typename VectorT>
151 unsigned VL = V.getLength();
159 for (
unsigned i = 1; i < VL; ++i)
160 if (V[i] != std::numeric_limits<PBQP::PBQPNum>::infinity())
178 template <
typename GraphT,
typename StackT>
182 typedef typename GraphT::RawVector RawVector;
186 while (!stack.empty()) {
187 NodeId NId = stack.back();
190 RawVector v = G.getNodeCosts(NId);
196 if (G.getNodeMetadata(NId).wasConservativelyAllocatable())
198 "must have available register options");
201 for (
auto EId : G.adjEdgeIds(NId)) {
202 const Matrix& edgeCosts = G.getEdgeCosts(EId);
203 if (NId == G.getEdgeNode1Id(EId)) {
204 NodeId mId = G.getEdgeNode2Id(EId);
207 NodeId mId = G.getEdgeNode1Id(EId);
Represents a solution to a PBQP problem.
unsigned getSelection(GraphBase::NodeId nodeId) const
Get a node's selection.
Vector getRowAsVector(unsigned R) const
Returns the given row as a vector.
Vector getColAsVector(unsigned C) const
Returns the given column as a vector.
void applyR1(GraphT &G, typename GraphT::NodeId NId)
Reduce a node of degree one.
bool hasRegisterOptions(const VectorT &V)
Solution backpropagate(GraphT &G, StackT stack)
unsigned getRows() const
Return the number of rows in this matrix.
void setSelection(GraphBase::NodeId nodeId, unsigned selection)
Set the selection for a given node.
void applyR2(GraphT &G, typename GraphT::NodeId NId)