Lines Matching defs:copyOp
370 LogicalResult matchAndRewrite(nvgpu::DeviceAsyncCopyOp copyOp,
758 nvgpu::DeviceAsyncCopyOp copyOp, PatternRewriter &rewriter) const {
760 LLVM_DEBUG(DBGS() << "copyOp : " << copyOp << "\n");
763 copyOp.getSrc().template getDefiningOp<memref::SubViewOp>();
765 copyOp.getDst().template getDefiningOp<memref::SubViewOp>();
768 return rewriter.notifyMatchFailure(copyOp, "does not use subview ops for "
772 SmallVector<Value> srcindices(copyOp.getSrcIndices().begin(),
773 copyOp.getSrcIndices().end());
779 rewriter, copyOp.getLoc(), srcSubViewOp.getMixedOffsets(),
785 SmallVector<Value> dstindices(copyOp.getDstIndices().begin(),
786 copyOp.getDstIndices().end());
792 rewriter, copyOp.getLoc(), dstSubViewOp.getMixedOffsets(),
800 copyOp, nvgpu::DeviceAsyncTokenType::get(copyOp.getContext()),
801 (dstSubViewOp ? dstSubViewOp.getSource() : copyOp.getDst()),
803 (srcSubViewOp ? srcSubViewOp.getSource() : copyOp.getSrc()),
804 foldedSrcIndices, copyOp.getDstElements(), copyOp.getSrcElements(),
805 copyOp.getBypassL1Attr());