[Mlir-commits] [mlir] [mlir][vector] Update tests for xfer permutation lowering (4/N) (PR #127624)
Hugo Trachino
llvmlistbot at llvm.org
Thu Feb 20 02:13:13 PST 2025
================
@@ -385,31 +408,57 @@ func.func @xfer_read_minor_identitiy_bcast_dims_scalable(
return %res : vector<8x[4]x2x3xf32>
}
+// CHECK-LABEL: func.func @xfer_read_minor_identitiy_bcast_dims_with_mask
+// CHECK-SAME: %[[MEM:.*]]: memref<?x?x?x?xf32>
+// CHECK-SAME: %[[MASK:.*]]: vector<4x3xi1>
+// CHECK-SAME: %[[IDX:.*]]: index) -> vector<8x4x2x3xf32>
+// CHECK: %[[PASS_THROUGH:.*]] = arith.constant 0.000000e+00 : f32
+// CHECK: %[[T_READ:.*]] = vector.transfer_read %[[MEM]][%[[IDX]], %[[IDX]], %[[IDX]], %[[IDX]]], %[[PASS_THROUGH]], %[[MASK]]{{.*}} permutation_map = #[[$MAP]]} : memref<?x?x?x?xf32>, vector<4x2x3xf32>
+// CHECK: %[[BC:.*]] = vector.broadcast %[[T_READ]] : vector<4x2x3xf32> to vector<8x4x2x3xf32>
+// CHECK: return %[[BC]] : vector<8x4x2x3xf32>
+func.func @xfer_read_minor_identitiy_bcast_dims_with_mask(
----------------
nujaa wrote:
Does this test deserve a scalable flavour ?
https://github.com/llvm/llvm-project/pull/127624
More information about the Mlir-commits
mailing list