nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
69 stars 30 forks source link

[CI] Add new shape and with/without ukernel for benchmark tracking #909

Closed newling closed 12 hours ago

newling commented 3 days ago

Now prints performance numbers for shapes

(M, N, K) = (512, 512, 4096) and (512, 4096, 512) and (4096, 512, 512)

with and without ukernels. Numbers with ukernels are interesting, take a look in CI, different gain with ukernel for the different shapes.

But looks like there's some improvement we can make on the vectorization of matmul front.

newling commented 2 days ago

Any idea of the windows failure?

Yes I'm 90% sure it'll be fixed after rebasing on https://github.com/nod-ai/iree-amd-aie/pull/911 (windows + ukernel = doesn't work)