nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
64 stars 29 forks source link

Prepare for testing non-matmul workloads #443

Closed newling closed 3 months ago

newling commented 3 months ago

Addresses the main aspect of https://github.com/nod-ai/iree-amd-aie/pull/394 that I found potentially difficult to use, namely that a single function (run_test) did 2 different things: supports matmul specific mlir file generation via flags like --m --n --k, but also accepts generic mlir files which have nothing to do with matmul. This PR makes this test infrastructure easier to scale IMO. See discussions in https://github.com/nod-ai/iree-amd-aie/pull/394