[Mlir-commits] [mlir] [MLIR] extend `getCompressedMaskOp` support in `VectorEmulateNarrowType` (PR #116122)

Andrzej WarzyƄski llvmlistbot at llvm.org
Thu Nov 14 09:34:46 PST 2024


================
@@ -104,10 +104,14 @@ static FailureOr<Operation *> getCompressedMaskOp(OpBuilder &rewriter,
   if (createMaskOp) {
     OperandRange maskOperands = createMaskOp.getOperands();
     size_t numMaskOperands = maskOperands.size();
+    // The `vector.create_mask` op creates a mask arrangement without any zeros
+    // at the front. Also, because `numFrontPadElems` is strictly smaller than
+    // `numSrcElemsPerDest`, the compressed mask generated by shifting the
+    // original mask by `numFrontPadElems` will not have any zeros at the front
+    // as well.
----------------
banach-space wrote:

I don't quite follow this comment. Is it meant to explain where the definition of `s0` is coming from? Or, more specifically:

> the compressed mask generated by shifting the original mask

where does shifting happen?

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


More information about the Mlir-commits mailing list