From 711f2ea0842908647be91f6f5bce91204e78af2a Mon Sep 17 00:00:00 2001 From: Kristof Denolf Date: Mon, 15 Apr 2024 13:16:12 -0600 Subject: [PATCH] reformat --- .../basic/vector_max_reduce/aie2.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/programming_examples/basic/vector_max_reduce/aie2.py b/programming_examples/basic/vector_max_reduce/aie2.py index 8c601a8686..068f53faca 100755 --- a/programming_examples/basic/vector_max_reduce/aie2.py +++ b/programming_examples/basic/vector_max_reduce/aie2.py @@ -15,6 +15,7 @@ import sys + def my_max_reduce(): N = 1024 @@ -38,8 +39,12 @@ def device_body(): memRef_O_ty = T.memref(1, T.i32()) # AIE Core Function declarations - max_reduce_vector = external_func("vector_max", inputs=[memRef_I_ty, memRef_O_ty, T.i32()]) - max_reduce_scalar = external_func("scalar_max", inputs=[memRef_I_ty, memRef_O_ty, T.i32()]) + max_reduce_vector = external_func( + "vector_max", inputs=[memRef_I_ty, memRef_O_ty, T.i32()] + ) + max_reduce_scalar = external_func( + "scalar_max", inputs=[memRef_I_ty, memRef_O_ty, T.i32()] + ) # Tile declarations ShimTile = tile(int(sys.argv[2]), 0) @@ -47,7 +52,9 @@ def device_body(): # AIE-array data movement with object fifos of_in = object_fifo("in", ShimTile, ComputeTile2, buffer_depth, memRef_I_ty) - of_out = object_fifo("out", ComputeTile2, ShimTile, buffer_depth, memRef_O_ty) + of_out = object_fifo( + "out", ComputeTile2, ShimTile, buffer_depth, memRef_O_ty + ) # Set up compute tiles