add verification to acquire op

Xilinx · Dec 20, 2023 · 7bd6f27 · 7bd6f27
1 parent 4485703
commit 7bd6f27
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/Dialect/AIE/IR/AIEDialect.cpp b/lib/Dialect/AIE/IR/AIEDialect.cpp
@@ -698,6 +698,16 @@ LogicalResult ObjectFifoAcquireOp::verify() {
           "on non-consumer tile");
   }
 
+  auto objFifoElem =
+    getObjectFifo().getElemType().cast<AIEObjectFifoType>().getElementType();
+  auto objFifoSubviewElem =
+      getResult().getType().cast<AIEObjectFifoSubviewType>().getElementType();
+  if (objFifoElem != objFifoSubviewElem)
+    return emitOpError(
+        "ObjectFifo element and ObjectFifoSubview elements must match.\n");
+
+  return success();
+
   return success();
 }
 

diff --git a/test/python/ipu.py b/test/python/ipu.py
@@ -943,7 +943,7 @@ def core_body():
                     ObjectFifoPort.Consume, "in1", 1, T.memref(8, T.i32())
                 ).acquired_elem()
                 elem_out = acquire(
-                    ObjectFifoPort.Produce, "out1", 1, T.memref(8, T.i32())
+                    ObjectFifoPort.Produce, "out1", 1, T.memref(16, T.i32())
                 ).acquired_elem()
                 for i in range_(8):
                     v0 = memref.load(elem_in, [i])