search

iree-org / iree-turbine

IREE's PyTorch Frontend, based on Torch Dynamo.
Apache License 2.0
55 stars 25 forks source link

Add support for dynamic parallel dims in GEMMs #274

Open harsh-nod opened 1 day ago

harsh-nod commented 1 day ago

This PR adds tests for dynamic M and N dims in GEMMs. This works out of the box for the most part and just requires moving the align index pass after scheduling.