tree 5a45b73f8bb6ba8d9232cfd622cb92be61ec2863
parent e6fb1e180438c4f78a1ea2bafd9a653fbe7a064b
author Daniel Garvey <danimal197@gmail.com> 1756226972 -0400
committer GitHub <noreply@github.com> 1756226972 -0500
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJoreWcCRC1aQ7uu5UhlAAAg6MQAInqyDILM5tt8qesO7T/aL1t
 3R8/0pheeIp0oe1hxodPuBDeBkLKzTWA1rX3OztVBtp8Yl0JBjbxaBfYU3tkawx9
 anhnwIWP5JYorKGADcO9Htc00EpjM/GrMgv6/HwMNpI4Nw55+EYj7RsSkUPxscBf
 xStwcJAfP6GhzzdJiBAhPwakd8oBuMa81XQohHOJE+VgoWEy/JcpLOWeTxjQn85y
 JEH/isxkBnQDutcHKycoT2TpT/+2J80pjb2+tznv/YWN+UoBA7+MGgFmDJxw7+Kg
 eQfnFza+aZ082JbBXdZsctgl2L6/y7g2qi1H2v0kvhPHyD1slYAdCgQSEWMUouYy
 1OAXZAhAFmsNFiSat6rW2b5DVHvri2c+fNluDRf/T+t5L1o4RU6H7i7QlLqo3sF9
 V6GQPWl08iCzVHmR4kMW241Y3ToHjVPhq1WbljSn2vS6skyiUvZozT7q0cw1CEDx
 Mod3BUx6Kq7XVakWviUwBJuAEDxtMmlJipaSFHXETMlMUqgwsRM0Agbkqag34rO8
 KcppQKOUQpIE5CC2VW31lyWFvW21AdUnp2BKUkowKv7xmyBO6YvasGXhqGPPSbIG
 q2JHHgMH5r8s5FZtMaK0SIRgrR/EIxRbxWEzZ27UdFhZs1q7D0bGf0OUgLEcqcre
 7TIdJ6oAkMhpgw+felot
 =QfVN
 -----END PGP SIGNATURE-----
 

[DispatchCreation] Fix trailing unit dims case for collapse of expand folding (#21677)

Previous logic only prevented collapsing all unit dims in a
reassociation element if the first element of the reassociation
represented a unit dim in the input.

i.e. it worked for cases like 1x1x44x5 -> 1x44x5
but failed for 5x44x1x1 -> 5x44x1

---------

Signed-off-by: dan <danimal197@gmail.com>
Signed-off-by: Ian Wood <ianwood@u.northwestern.edu>
Co-authored-by: Ian Wood <ianwood@u.northwestern.edu>