[GPU] Use affine.delinearize_index for MMA tiles and vector distribution

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

http://iree.dev/

Apache License 2.0

2.85k stars 621 forks source link

Open krzysz00 opened 5 days ago

krzysz00 commented 5 days ago

Commit message TBD, tests not yet written, still putting together some canonicalizations looking an the intermediates revealed