[Mlir-commits] [mlir] [mlir] [vector] Add linearization pattern for vector.create_mask (PR #138214)
Nishant Patel
llvmlistbot at llvm.org
Mon May 12 19:52:27 PDT 2025
================
@@ -445,6 +445,64 @@ struct LinearizeVectorSplat final
}
};
+/// This pattern converts the CreateMaskOp to work on a linearized vector.
+/// It currently supports only 2D masks with a unit outer dimension.
+/// Following,
+/// vector.create_mask %arg0, %arg1 : vector<1x4xi1>
+/// is converted to:
+/// %zero = arith.constant 0 : index
+/// %cmpi = arith.cmpi sgt, %arg0, %zero : index
+/// %index = arith.index_cast %cmpi : i1 to index
+/// %mul = arith.muli %index, %arg1 : index
----------------
nbpatel wrote:
oh yes, good catch. Thanks
https://github.com/llvm/llvm-project/pull/138214
More information about the Mlir-commits
mailing list