llvm-project
9f0a912a - [mlir][test][sve] Add e2e test for linalg.pack + linalg.unpack (#129696)

Commit
1 year ago
[mlir][test][sve] Add e2e test for linalg.pack + linalg.unpack (#129696) This patch adds an e2e test for the `linalg.pack` + `linalg.unpack` pair with a dynamic inner tile size that's tied to SVE's "vscale": ```mlir %c4 = arith.constant 4 : index %vs = vector.vscale %tile_size = arith.muli %c4, %vs : index ``` This means that the actual size of the corresponding inner and outer tile size will depend on the runtime value of "vscale". To make the new test deterministic (and to make it easier to experiment), I have hard-coded the value of "vscale" to 2 via (2 x 128 bits = 256 bits): ```mlir `func.call @setArmVLBits(%c256) : (i32) -> () ``` This can be relaxed at a later time or played with when experimenting locally with e.g. QEMU. NOTE: Vectorization has not been enabled yet (scalable vectorization of `linalg.unpack` is still WIP).
Author
Parents
Loading