[Mlir-commits] [mlir] [MLIR][Vector] Allow any strided memref for one-element vector.load in lowering vector.gather (PR #122437)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Fri Jan 10 11:28:37 PST 2025
================
@@ -136,6 +136,24 @@ func.func @gather_tensor_1d(%base: tensor<?xf32>, %v: vector<2xindex>, %mask: ve
return %0 : vector<2xf32>
}
+// CHECK-LABEL: @gather_strided_memref_1d
+// CHECK: %[[MASK:.*]] = vector.extract %arg2[0] : i1 from vector<1xi1>
+// CHECK: %1 = vector.extract %arg1[0] : index from vector<1xindex>
----------------
PragmaTwice wrote:
Ahh sorry for missing that. Fixed.
https://github.com/llvm/llvm-project/pull/122437
More information about the Mlir-commits
mailing list