Lines Matching defs:writeOp

379                                 vector::TransferWriteOp writeOp,
381     VectorType writtenVectorType = writeOp.getVectorType();
389     AffineMap map = distributionMapFn(writeOp.getVector());
397     if (writeOp.getMask()) {
404       if (!writeOp.getPermutationMap().isMinorIdentity())
407           getDistributedType(writeOp.getMaskType(), map, warpOp.getWarpSize());
413         cloneWriteOp(rewriter, warpOp, writeOp, targetType, maskType);
467                              vector::TransferWriteOp writeOp,
469     Location loc = writeOp.getLoc();
470     VectorType vecType = writeOp.getVectorType();
485     SmallVector<Value> yieldValues = {writeOp.getVector()};
492     // Create a second warp op that contains only writeOp.
498         cast<vector::TransferWriteOp>(rewriter.clone(*writeOp.getOperation()));
500     rewriter.eraseOp(writeOp);
510     auto writeOp = dyn_cast_or_null<vector::TransferWriteOp>(lastNode);
511     if (!writeOp)
514     Value maybeMask = writeOp.getMask();
515     if (!llvm::all_of(writeOp->getOperands(), [&](Value value) {
516           return writeOp.getVector() == value ||
522     if (succeeded(tryDistributeOp(rewriter, writeOp, warpOp)))
526     if (writeOp.getMask())
529     if (succeeded(tryExtractOp(rewriter, writeOp, warpOp)))
536   /// Clone `writeOp` assumed to be nested under `warpOp` into a new warp
538   /// write the result of the new warp execute op. The old `writeOp` is deleted.
541                                        vector::TransferWriteOp writeOp,
544     assert(writeOp->getParentOp() == warpOp &&
551           rewriter, warpOp, ValueRange{writeOp.getVector(), writeOp.getMask()},
555           rewriter, warpOp, ValueRange{{writeOp.getVector()}},
560         cast<vector::TransferWriteOp>(rewriter.clone(*writeOp.getOperation()));
561     rewriter.eraseOp(writeOp);