[Mlir-commits] [mlir] [mlir][Vector] Fix out-of-bounds crash unrolling create_mask/constant_mask with rank-mismatched native shape (PR #217514)

[Aman Singh]

================
@@ -1162,18 +1188,24 @@ struct UnrollConstantMaskPattern
 
     VectorType resultType = constantMaskOp.getVectorType();
     SmallVector<int64_t> originalSize = *constantMaskOp.getShapeForUnroll();
+    // The logic below assumes a 1-1 correspondence between the unroll shape and
+    // the op's mask dimensions, so pad the target shape with leading unit
+    // dimensions when it has a smaller rank than the op (as is done in
+    // `UnrollElementwisePattern`).
----------------
amanyagami wrote:

Applied in `c521ec4b`.

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