commit 4dd50376d4e1c0bbbbe998de14aad7028518ebed
author Kojo Acquah <KoolJBlack@users.noreply.github.com> Fri Apr 19 07:49:25 2024 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Apr 19 07:56:12 2024 -0700
tree c0004cdd2fb27f5ba57d4ce5ce9fbac7132426db
parent 482a751f2200bb343988f1383fd175b8e967026b

[mlir][ArmNeon] Update `LowerContractionToSMMLAPattern` to support proper unrolling for k dimension (#88591)

Fixes correctness issue with current smmla unrolling patterns whereby
unrolling K dimension would only include the result from the last tile
along K. Updates patterns to feed previous smmla output of the previous
tile into the next one along K.

GitOrigin-RevId: b859a9fc152f83d657b1b59a99ede2478e285b27