nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
46 stars 23 forks source link

Improve support for the microkernel matmul path #435

Open newling opened 2 weeks ago

newling commented 2 weeks ago

This task description needs improvement but essentially we should do these in order:

1) Improve error messages for cases we don't currently support (example: inner-most tiling of operands not having 64 elements each),

2) improve set of supported cases: different strides, smaller tiles in core memory, etc.

This is related to issue: https://github.com/Xilinx/mlir-aie/issues/1478

Currently we have to be very careful when tiling our matmul that the inner loop has the right matmul shape and striding, otherwise the numerics go haywire.