Do not always assign buffers to mem_bank 0 (Xilinx#759)

nod-ai · Nov 4, 2024 · dcabc84 · dcabc84
1 parent 9a48bfb
commit dcabc84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlir/lib/Conversion/AIRToAIEPass.cpp b/mlir/lib/Conversion/AIRToAIEPass.cpp
@@ -186,7 +186,7 @@ AIE::BufferOp allocateBufferOp(uint64_t &BufferId, MemRefType memrefTy,
   AIE::BufferOp bufferOp = builder.create<AIE::BufferOp>(
       tile->getLoc(), memrefTy, tile, /*sym_name*/ nullptr,
       /*address*/ nullptr, /*initial_value*/ nullptr,
-      /*mem_bank*/ builder.getI32IntegerAttr(0));
+      /*mem_bank*/ nullptr);
 
   std::stringstream ss =
       generateBufferNameInStringStream("buf", BufferId, attr, x, y);