nod-ai / iree-amd-aie

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

Support depth-wise and 2D (dense) convolution #439

Open newling opened 2 weeks ago

newling commented 2 weeks ago

Eventually we should able to use any of the following approaches: winogram, igemm, and direct. With the ability to fuse elementwise operations which follow, full parity with matmul.