Skip to content
GitLab
Unverified Commit b859a9fc authored by Kojo Acquah's avatar Kojo Acquah Committed by GitHub
Browse files

[mlir][ArmNeon] Update `LowerContractionToSMMLAPattern` to support proper... 

[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.
parent f4c0c40f
Expand all Hide whitespace changes
Inline Side-by-side
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment