iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.85k stars 620 forks source link

Guard the blocking of dynamic dimensions of contractions. #19164

Closed MaheshRavishankar closed 1 week ago

MaheshRavishankar commented 1 week ago

The change to block dynamic dimensions of contractions based on divisibility analysis seems to have regressed quite a few tests. The change is probably too aggressive. Guarding that by a flag while I fix those forward.