Home
last modified time | relevance | path

Searched refs:copyOp (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1165 memref::CopyOp copyOp; in matchAndRewrite() local
1174 copyOp = newCopyOp; in matchAndRewrite()
1178 if (!copyOp) in matchAndRewrite()
1180 LDBG("with copy " << *copyOp); in matchAndRewrite()
1204 Value in = copyOp.getSource(); in matchAndRewrite()
1218 rewriter.eraseOp(copyOp); in matchAndRewrite()
1245 memref::CopyOp copyOp; in matchAndRewrite() local
1252 copyOp = newCopyOp; in matchAndRewrite()
1256 if (!copyOp) in matchAndRewrite()
1261 Value out = copyOp.getTarget(); in matchAndRewrite()
[all …]
H A DTransforms.cpp630 memref::CopyOp copyOp, PatternRewriter &rewriter) const { in matchAndRewrite() argument
631 return vectorizeCopy(rewriter, copyOp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp105 if (auto copyOp = dyn_cast<DeviceAsyncCopyOp>(op)) in getIndices() local
106 return copyOp.getDstIndices(); in getIndices()
121 if (auto copyOp = dyn_cast<DeviceAsyncCopyOp>(op)) in setIndices() local
122 return copyOp.getDstIndicesMutable().assign(indices); in setIndices()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DMultiBuffer.cpp23 auto copyOp = dyn_cast<memref::CopyOp>(op); in overrideBuffer() local
24 if (!copyOp) in overrideBuffer()
26 return copyOp.getTarget() == buffer; in overrideBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp650 LogicalResult matchAndRewrite(CopyOp copyOp, in matchAndRewrite()
655 if (auto castOp = copyOp.getSource().getDefiningOp<CastOp>()) { in matchAndRewrite()
662 rewriter.updateRootInPlace(copyOp, [&] { in matchAndRewrite()
663 copyOp.getSourceMutable().assign(castOp.getSource()); in matchAndRewrite()
671 if (auto castOp = copyOp.getTarget().getDefiningOp<CastOp>()) { in matchAndRewrite()
678 rewriter.updateRootInPlace(copyOp, [&] { in matchAndRewrite()
679 copyOp.getTargetMutable().assign(castOp.getSource()); in matchAndRewrite()
694 LogicalResult matchAndRewrite(CopyOp copyOp, in matchAndRewrite()
696 if (copyOp.getSource() != copyOp.getTarget()) in matchAndRewrite()
699 rewriter.eraseOp(copyOp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h309 LogicalResult vectorizeCopy(RewriterBase &builder, memref::CopyOp copyOp);
974 LogicalResult matchAndRewrite(memref::CopyOp copyOp,