Simple bank-aware malloc scheme (#1143)

Co-authored-by: AndraBisca <andrab@amd.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Xilinx · Apr 25, 2024 · 37dd2a6 · 37dd2a6
1 parent a26185e
commit 37dd2a6
Show file tree

Hide file tree

Showing 45 changed files with 551 additions and 106 deletions.
diff --git a/include/aie/Dialect/AIE/IR/AIEOps.td b/include/aie/Dialect/AIE/IR/AIEOps.td
@@ -1258,7 +1258,8 @@ def AIE_BufferOp: AIE_Op<"buffer", [
     ins Index:$tile,
     OptionalAttr<StrAttr>:$sym_name,
     OptionalAttr<AIEI32Attr>:$address,
-    OptionalAttr<ElementsAttr>:$initial_value
+    OptionalAttr<ElementsAttr>:$initial_value,
+    OptionalAttr<AIEI32Attr>:$mem_bank
   );
 
   let results = (outs AnyMemRef:$buffer);

diff --git a/include/aie/Dialect/AIE/Transforms/AIEPasses.td b/include/aie/Dialect/AIE/Transforms/AIEPasses.td
@@ -25,6 +25,11 @@ def AIEAssignBufferAddresses : Pass<"aie-assign-buffer-addresses", "DeviceOp"> {
   }];
 
   let constructor = "xilinx::AIE::createAIEAssignBufferAddressesPass()";
+
+  let options = [
+    Option<"clBasicAlloc", "basic-alloc", "bool", /*default=*/"false",
+            "Flag to enable the basic sequential allocation scheme (not bank-aware).">
+  ];
 }
 
 def AIEAssignLockIDs : Pass<"aie-assign-lock-ids", "DeviceOp"> {