[Mlir-commits] [mlir] [Linalg][Vectorization] Add support for linalg vectorization of a tensor.extract case (PR #107922)

Nirvedh Meshram llvmlistbot at llvm.org
Thu Sep 19 14:33:48 PDT 2024


nirvedhmeshram wrote:

@banach-space https://github.com/llvm/llvm-project/pull/100582/files breaks the logic here as one of the test cases is
```
%5 = linalg.generic {indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>], iterator_types = ["parallel", "parallel"]} outs(%arg1 : tensor<?x?xf32>) {
^bb0(%out: f32):
  %6 = linalg.index 1 : index
  %7 = arith.addi %6, %arg2 : index
  %extracted = tensor.extract %arg0[%c79, %7] : tensor<?x?xf32>
  linalg.yield %extracted : f32
} -> tensor<?x?xf32>
```
where it will find two non unit dims and hit the assert


https://github.com/llvm/llvm-project/pull/107922


More information about the Mlir-commits mailing list