Zero padding on MemTiles

[pvasireddy-amd]

This PR introduces the ability to apply zero padding on MemTiles in 3 dimensions. Specifically, it allows padding to be specified at the object FIFO level for MemTile objects through the addition of a padDimensions attribute.

In python code:

of_out = object_fifo(
    "out",
    MemTile,
    ShimTile,
    1,
    memRef_ty2,
    dimensionsToStream=[(5, 5), (5, 5)],
    padDimensions=[(2, 0), (3, 0)],
) 

Lowers to:

aie.objectfifo @out(%tile_0_1 dimensionsToStream [<size = 5, stride = 5>, <size = 5, stride = 5>], {%tile_0_0}, 1 : i32) 
{padDimensions = #aie<bd_pad_layout_array[<const_pad_before = 2, const_pad_after = 0>, <const_pad_before = 3, const_pad_after = 0>]>} : !aie.objectfifo<memref<56xi32>>

Padding can also be applied in the runtime sequence using writebds or through dma_bd_chain.

[github-actions]

Coverage Report

Created: 2024-11-18 17:50

Click here for information about interpreting this report.

Filename	Function Coverage	Line Coverage	Region Coverage	Branch Coverage
AIE/IR/AIEDialect.cpp	91.11%	82.63%	85.16%	75.23%
AIE/Transforms/AIEObjectFifoStatefulTransform.cpp	100.00%	93.32%	90.41%	84.87%
AIEX/IR/AIEXDialect.cpp	100.00%	86.21%	89.26%	79.70%
AIEX/Transforms/AIECtrlPacketToDma.cpp	100.00%	96.64%	80.00%	68.18%
AIEX/Transforms/AIEDMATasksToNPU.cpp	86.36%	83.53%	87.91%	83.33%
AIEX/Transforms/AIEDmaToNpu.cpp	96.15%	89.18%	76.80%	60.48%
Totals	93.92%	87.68%	86.62%	78.14%

Generated by llvm-cov -- llvm version 14.0.0

[pvasireddy-amd]

To DO: Cleanup and moving checks to verifier in AIEDMATasksToNpu

[AndraBisca]

My latest PR that was merged into main has some updates to the documentation and objectfifo python API. Could you please add some similar style documentation for the mem tile padding in the programming guide as well? I think somewhere at the end of Section 2c should be good. This can be a different PR btw, after this one is merged.

[AndraBisca]

Looks good! Just these comments to address and it should be good to merge.

[AndraBisca]

Typo?

[AndraBisca]

typo?