Skip to content

Commit c114a89

Browse files
qedawkinsclaude
andauthored
[PCF] Add tensor.collapse_shape fusion into pcf.generic/loop (#23491)
Fuse tensor.collapse_shape consumers into producer pcf.generic/pcf.loop ops. This supports all possible dynamic cases by emitting loops along outer dimensions, effectively unrolling the producer `pcf.write_slice` ops allowing for continued fusion. The alternative to this (essentially premature unrolling) is to put the reshape on the sref, though to continue fusing we still would have to do the same (or propagate the collapse, which should have already been attempted by this point). --------- Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
1 parent a02e85f commit c114a89

6 files changed

Lines changed: 966 additions & 41 deletions

File tree

0 commit comments

Comments
 (0)