Subzero. Enables moar complex relocation offsets.

src/IceOperand.cpp

 //===- subzero/src/IceOperand.cpp - High-level operand implementation -----===//
 //
 //                        The Subzero Code Generator
 //
 // This file is distributed under the University of Illinois Open Source
 // License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 ///
 /// \file
 /// \brief Implements the Operand class and its target-independent subclasses,
 /// primarily for the methods of the Variable class.
 ///
 //===----------------------------------------------------------------------===//
 
 #include "IceOperand.h"
 
 #include "IceCfg.h"
 #include "IceCfgNode.h"
 #include "IceInst.h"
 #include "IceInstVarIter.h"
 #include "IceTargetLowering.h" // dumping stack/frame pointer register
 
 namespace Ice {
 
 bool operator==(const RelocatableTuple &A, const RelocatableTuple &B) {
-  return A.Offset == B.Offset && A.Name == B.Name;
+  // A and B are the same if:
+  //   (1) they have the same name; and
+  //   (2) they have the same offset.
+  //
+  // (1) is trivial to check, but (2) requires some care.
+  //
+  // For (2):
+  //  if A and B have known offsets (i.e., no symbolic references), then
+  //     A == B -> A.Offset == B.Offset.
+  //  else each element i in A.OffsetExpr[i] must be the same (or have the same
+  //    value) as B.OffsetExpr[i].
+  if (A.Name != B.Name) {
+    return false;
+  }
+
+  bool BothHaveKnownOffsets = true;
+  RelocOffsetT OffsetA = 0;
+  RelocOffsetT OffsetB = 0;
+  for (SizeT i = 0; i < A.OffsetExpr.size() && BothHaveKnownOffsets; ++i) {
+    BothHaveKnownOffsets = A.OffsetExpr[i]->hasOffset();
+    if (BothHaveKnownOffsets) {
+      OffsetA += A.OffsetExpr[i]->getOffset();
+    }
+  }
+  for (SizeT i = 0; i < B.OffsetExpr.size() && BothHaveKnownOffsets; ++i) {
+    BothHaveKnownOffsets = B.OffsetExpr[i]->hasOffset();
+    if (BothHaveKnownOffsets) {
+      OffsetB += B.OffsetExpr[i]->getOffset();
+    }
+  }
+  if (BothHaveKnownOffsets) {
+    // Both have known offsets (i.e., no unresolved symbolic references), so
+    // A == B -> A.Offset == B.Offset.
+    return OffsetA == OffsetB;
+  }
+
+  // Otherwise, A and B are not the same if their OffsetExpr's have different
+  // sizes.
+  if (A.OffsetExpr.size() != B.OffsetExpr.size()) {
+    return false;
+  }
+
+  // If the OffsetExprs' sizes are the same, then
+  // for each i in OffsetExprSize:
+  for (SizeT i = 0; i < A.OffsetExpr.size(); ++i) {
+    const auto *const RelocOffsetA = A.OffsetExpr[i];
+    const auto *const RelocOffsetB = B.OffsetExpr[i];
+    if (RelocOffsetA->hasOffset() && RelocOffsetB->hasOffset()) {
+      // A.OffsetExpr[i].Offset == B.OffsetExpr[i].Offset iff they are both
+      // defined;
+      if (RelocOffsetA->getOffset() != RelocOffsetB->getOffset()) {
+        return false;
+      }
+    } else if (RelocOffsetA != RelocOffsetB) {
+      // or, if they are undefined, then the RelocOffsets must be the same.
+      return false;
+    }
+  }
+
+  return true;
 }
 
 bool operator<(const RegWeight &A, const RegWeight &B) {
   return A.getWeight() < B.getWeight();
 }
 bool operator<=(const RegWeight &A, const RegWeight &B) { return !(B < A); }
 bool operator==(const RegWeight &A, const RegWeight &B) {
   return !(B < A) && !(A < B);
 }
 
@@ skipping 456 matching lines @@
 
 void ConstantRelocatable::dump(const Cfg *Func, Ostream &Str) const {
   if (!BuildDefs::dump())
     return;
   Str << "@";
   if (Func && !SuppressMangling) {
     Str << Func->getContext()->mangleName(Name);
   } else {
     Str << Name;
   }
-  if (Offset)
-    Str << "+" << Offset;
+  const RelocOffsetT Offset = getOffset();
+  if (Offset) {
+    if (Offset >= 0) {
+      Str << "+";
+    }
+    Str << Offset;
+  }
 }
 
 void ConstantUndef::emit(TargetLowering *Target) const { Target->emit(this); }
 
 void LiveRange::dump(Ostream &Str) const {
   if (!BuildDefs::dump())
     return;
   bool First = true;
   for (const RangeElementType &I : Range) {
     if (!First)
@@ skipping 31 matching lines @@
   if (getType() != IceType_i32 && getType() != IceType_i16 &&
       getType() != IceType_i8)
     return false;
   // The Following checks if the signed representation of Value is between
   // -Threshold/2 and +Threshold/2
   bool largerThanThreshold = Threshold / 2 + Value >= Threshold;
   return largerThanThreshold;
 }
 
 } // end of namespace Ice