Update ipu.py

Xilinx · Dec 20, 2023 · 772b357 · 772b357
1 parent 581f14e
commit 772b357
Show file tree

Hide file tree

Showing 13 changed files with 447 additions and 311 deletions.
diff --git a/reference_designs/ipu-xrt/add_one_objFifo/aie2.py b/reference_designs/ipu-xrt/add_one_objFifo/aie2.py
@@ -73,7 +73,9 @@ def sequence(inTensor, notUsed, outTensor):
                 ipu_dma_memcpy_nd(
                     metadata="in0", bd_id=1, mem=inTensor, lengths=[1, 1, 1, 64]
                 )
-                ipu_sync(column=0, row=0, direction=0, channel=0)
+                ipu_sync(
+                    column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/log_hello_world/hello_world.py b/reference_designs/ipu-xrt/log_hello_world/hello_world.py
@@ -63,7 +63,9 @@ def sequence(in_mem, out_mem, logout):
                 ipu_dma_memcpy_nd(
                     metadata="logoutOF", bd_id=2, mem=logout, lengths=[1, 1, 1, N]
                 )
-                ipu_sync(column=0, row=0, direction=0, channel=0)
+                ipu_sync(
+                    column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/matrix_multiplication/aie2.py b/reference_designs/ipu-xrt/matrix_multiplication/aie2.py
@@ -217,7 +217,9 @@ def sequence(A, B, C):
                             strides=[n_in_i32s, k_x_N_in_i32s, N_in_i32s],
                         )
 
-                    ipu_sync(column=0, row=0, direction=0, channel=0)
+                    ipu_sync(
+                        column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                    )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/matrix_multiplication_column/aie2.py b/reference_designs/ipu-xrt/matrix_multiplication_column/aie2.py
@@ -250,7 +250,9 @@ def sequence(A, B, C):
                             strides=[0, n_in_i32s, N_in_i32s],
                         )
 
-                    ipu_sync(column=0, row=0, direction=0, channel=0)
+                    ipu_sync(
+                        column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                    )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/passthrough_hardware/aie2.py b/reference_designs/ipu-xrt/passthrough_hardware/aie2.py
@@ -53,7 +53,9 @@ def core_body():
             def sequence(A, B, C):
                 ipu_dma_memcpy_nd(metadata="out", bd_id=0, mem=C, lengths=[1, 1, 1, N])
                 ipu_dma_memcpy_nd(metadata="in", bd_id=1, mem=A, lengths=[1, 1, 1, N])
-                ipu_sync(column=0, row=0, direction=0, channel=0)
+                ipu_sync(
+                    column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/vector_scalar/aie2.py b/reference_designs/ipu-xrt/vector_scalar/aie2.py
@@ -70,7 +70,9 @@ def core_body():
             def sequence(A, B, C):
                 ipu_dma_memcpy_nd(metadata="out", bd_id=0, mem=C, lengths=[1, 1, 1, N])
                 ipu_dma_memcpy_nd(metadata="in", bd_id=1, mem=A, lengths=[1, 1, 1, N])
-                ipu_sync(column=0, row=0, direction=0, channel=0)
+                ipu_sync(
+                    column=0, row=0, direction=0, channel=0, column_num=1, row_num=1
+                )
 
     print(ctx.module)
 

diff --git a/reference_designs/ipu-xrt/vision_pipelines/color_detect/aie2_colorDetect.py b/reference_designs/ipu-xrt/vision_pipelines/color_detect/aie2_colorDetect.py
@@ -13,6 +13,7 @@
 from aie.extras.dialects.ext.scf import for_
 from aie.extras.context import mlir_mod_ctx
 from aie.extras import types as T
+from aie.ir import TypeAttr
 
 
 width = 64