Fix when ops have multiple operands

xmos · May 14, 2024 · 0c200df · 0c200df
1 parent 8813815
commit 0c200df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xformer/Analysis/MemoryPlan.cpp b/xformer/Analysis/MemoryPlan.cpp
@@ -167,9 +167,9 @@ std::vector<int> MemoryPlan::getAllocatedOffsets(const bool overlapOps,
         auto outVal = o->getResult(0);
         auto nextOp = *outVal.getUsers().begin();
         // Identify chain of overlappable Ops
-        while (outVal.hasOneUse() &&
+        while (outVal.hasOneUse() && !alreadyVisited.contains(nextOp) &&
                nextOp->hasTrait<OpTrait::xcore::MemoryOverlappable>()) {
-          inVal = nextOp->getOperand(0);
+          inVal = outVal;
           inputVals.push_back(inVal);
           alreadyVisited.insert(nextOp);
           outVal = nextOp->getResult(0);