Change tiling and packing sizes to make outer dimension 2x2

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator

Apache License 2.0

46 stars 23 forks source link

Closed yzhang93 closed 2 weeks ago

yzhang93 commented 2 weeks ago

This PR changes the packing strategy to offload the large inner tile sizes to outer dimensions. This avoids the large allocations in L1, and is required in the objectFifo pipeline DistributeCoresAndObjectFifos pass.
This PR also bumps the MLIR-AIR to include a fix to support the new packing strategy.