Home
last modified time | relevance | path

Searched refs:transferWriteOp (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DHoisting.cpp47 vector::TransferWriteOp transferWriteOp; member
101 assert(write.transferWriteOp && in findMatchingTransferRead()
137 if (u == write.transferWriteOp) in findMatchingTransferRead()
264 assert(read.transferReadOp && write.transferWriteOp && in hoistReadWrite()
297 write.transferWriteOp->moveAfter(forOp); in hoistReadWrite()
310 return SmallVector<Value>{write.transferWriteOp.getVector()}; in hoistReadWrite()
323 write.transferWriteOp.getSourceMutable().assign( in hoistReadWrite()
329 .replaceAllUsesWith(write.transferWriteOp.getResult()); in hoistReadWrite()
330 write.transferWriteOp.getSourceMutable().assign( in hoistReadWrite()
359 if (!write.transferWriteOp || !write.transferWriteOp->hasOneUse()) in hoistRedundantVectorTransfersOnTensor()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp307 auto loc = transferWriteOp.getLoc(); in matchAndRewrite()
308 Value vector = transferWriteOp.getVector(); in matchAndRewrite()
310 Value source = transferWriteOp.getSource(); in matchAndRewrite()
318 if (transferWriteOp.hasOutOfBoundsDim()) in matchAndRewrite()
328 if (llvm::any_of(transferWriteOp.getIndices(), in matchAndRewrite()
477 auto loc = transferWriteOp.getLoc(); in matchAndRewrite()
478 Value vector = transferWriteOp.getVector(); in matchAndRewrite()
480 Value source = transferWriteOp.getSource(); in matchAndRewrite()
493 if (transferWriteOp.hasOutOfBoundsDim()) in matchAndRewrite()
497 if (transferWriteOp.getMask()) in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DFoldSubViewOps.cpp195 vector::TransferWriteOp transferWriteOp, memref::SubViewOp subViewOp, in replaceOp() argument
198 if (transferWriteOp.getTransferRank() == 0) in replaceOp()
201 transferWriteOp, transferWriteOp.getVector(), subViewOp.getSource(), in replaceOp()
204 transferWriteOp.getPermutationMap()), in replaceOp()
205 transferWriteOp.getInBoundsAttr()); in replaceOp()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp852 .Case([&](vector::TransferWriteOp transferWriteOp) { in convertVectorToNVVMCompatibleMMASync() argument
853 return convertTransferWriteToStores(transferWriteOp, in convertVectorToNVVMCompatibleMMASync()