Fixes to objFifo ops (#567)

* Fix lowering for objectFifos that have producer and consumer on the same core * Small fixes. Correctly identify shared tile in LinkOp when there is a broadcast objFifo. * Clang format and fix test output --------- Co-authored-by: abisca <[email protected]>
Xilinx · Aug 7, 2023 · 1cf0528 · 1cf0528
1 parent 109566c
commit 1cf0528
Show file tree

Hide file tree

Showing 5 changed files with 420 additions and 122 deletions.
diff --git a/lib/Dialect/AIE/IR/AIEDialect.cpp b/lib/Dialect/AIE/IR/AIEDialect.cpp
@@ -498,7 +498,7 @@ std::optional<Value> xilinx::AIE::ObjectFifoLinkOp::getOptionalSharedTile() {
         return {};
     }
     return {fifoOut.getProducerTile()};
-  } else {
+  } else if (isDistribute()) {
     auto fifoIn = getFifoIns()[0].getDefiningOp<ObjectFifoCreateOp>();
     for (auto fifoOut : getFifoOuts()) {
       ObjectFifoCreateOp fifoOutOp =
@@ -507,6 +507,13 @@ std::optional<Value> xilinx::AIE::ObjectFifoLinkOp::getOptionalSharedTile() {
         return {};
     }
     return {fifoIn.getConsumerTiles()[0]};
+  } else {
+    auto fifoIn = getFifoIns()[0].getDefiningOp<ObjectFifoCreateOp>();
+    auto fifoOut = getFifoOuts()[0].getDefiningOp<ObjectFifoCreateOp>();
+    for (auto consumerIn : fifoIn.getConsumerTiles())
+      if (consumerIn == fifoOut.getProducerTile())
+        return {fifoOut.getProducerTile()};
+    return {};
   }
   return {};
 }