Unverified Commit fe84369c authored by Kojo Acquah's avatar Kojo Acquah Committed by GitHub
Browse files

[mlir][ArmNeon] Implements unrolling patterns for LowerContractionToSMMLAPattern (#84848)

This patch updates `LowerContractionToSMMLAPattern` to unroll larger vector contracts into multiple smmla instructions. 

Now accepts up to [8,8,8] tiles (previously only [2,2,8]). The N/M dimensions must be powers of 2. `vector.extract_strided_slice`/`vector.insert_strided_slice` divides the contract into tiles to be processed in a row.
parent ab76052f
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