[Mlir-commits] [mlir] [mlir][vector] Update `CastAwayInsertLeadingOneDim` (PR #211777)

Federico Bruzzone llvmlistbot at llvm.org
Fri Aug 28 07:40:36 PDT 2026


================
@@ -594,11 +592,11 @@ func.func @cast_away_insert_leading_one_dims_rank2_scalable(%s: vector<1x[4]xf32
 
 // CHECK-LABEL: func @cast_away_insert_leading_one_dims_rank2_one_dest
 //  CHECK-SAME: (%[[S:.+]]: vector<1x4xf32>, %[[V:.+]]: vector<1x2x1x4xf32>)
-//       CHECK:   %[[EXTRACTS:.+]] = vector.extract %[[S]][0] : vector<4xf32> from vector<1x4xf32>
-//       CHECK:   %[[EXTRACTV:.+]] = vector.extract %[[V]][0] : vector<2x1x4xf32> from vector<1x2x1x4xf32>
-//       CHECK:   %[[INSERT:.+]] = vector.insert %[[EXTRACTS]], %[[EXTRACTV]] [1, 0] : vector<4xf32> into vector<2x1x4xf32>
-//       CHECK:   %[[BCAST:.+]] = vector.broadcast %[[INSERT]] : vector<2x1x4xf32> to vector<1x2x1x4xf32>
-//       CHECK:   return %[[BCAST]]
+//       CHECK:   %[[SCS:.+]] = vector.shape_cast %[[S]] : vector<1x4xf32> to  vector<4xf32>
----------------
FedericoBruzzone wrote:

```suggestion
//       CHECK:   %[[SCS:.+]] = vector.shape_cast %[[S]] : vector<1x4xf32> to vector<4xf32>
```

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


More information about the Mlir-commits mailing list