[Mlir-commits] [mlir] [mlir][vector] Add support for vector.maskedstore sub-type emulation. (PR #73871)
Diego Caballero
llvmlistbot at llvm.org
Thu Nov 30 03:08:51 PST 2023
================
@@ -32,6 +32,78 @@ using namespace mlir;
#define DBGSNL() (llvm::dbgs() << "\n")
#define LDBG(X) LLVM_DEBUG(DBGS() << X << "\n")
+/// Returns a compressed mask. The mask value is set only if any mask is present
+/// in the the scale range. E.g., if `scale` equals to 2, the following mask:
+///
+/// %mask = [1, 1, 1, 0, 0, 0]
+///
+/// will return the following new compressed mask:
+///
+/// %mask = [1, 1, 0]
+static FailureOr<Operation *> getCompressedMaskOp(OpBuilder &rewriter,
+ Location loc, Value mask,
+ int origElements, int scale) {
+ auto numElements = (origElements + scale - 1) / scale;
+
+ auto maskOp = mask.getDefiningOp();
----------------
dcaballe wrote:
nit: spell out `auto`
https://github.com/llvm/llvm-project/pull/73871
More information about the Mlir-commits
mailing list