Searched refs:transferReadOp (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferOpTransforms.cpp | 264 auto loc = transferReadOp.getLoc(); in matchAndRewrite()
265 Value vector = transferReadOp.getVector(); in matchAndRewrite()
267 Value source = transferReadOp.getSource(); in matchAndRewrite()
275 if (transferReadOp.hasOutOfBoundsDim()) in matchAndRewrite()
277 if (!transferReadOp.getPermutationMap().isMinorIdentity()) in matchAndRewrite()
285 if (llvm::any_of(transferReadOp.getIndices(), in matchAndRewrite()
419 auto loc = transferReadOp.getLoc(); in matchAndRewrite()
420 Value vector = transferReadOp.getVector(); in matchAndRewrite()
422 Value source = transferReadOp.getSource(); in matchAndRewrite()
435 if (transferReadOp.hasOutOfBoundsDim()) in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldSubViewOps.cpp | 171 vector::TransferReadOp transferReadOp, memref::SubViewOp subViewOp, in replaceOp() argument
174 if (transferReadOp.getTransferRank() == 0) in replaceOp()
177 transferReadOp, transferReadOp.getVectorType(), subViewOp.getSource(), in replaceOp()
180 transferReadOp.getPermutationMap()), in replaceOp()
181 transferReadOp.getPadding(), in replaceOp()
182 /*mask=*/Value(), transferReadOp.getInBoundsAttr()); in replaceOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Hoisting.cpp | 53 vector::TransferReadOp transferReadOp; member
173 if (user == candidateRead.transferReadOp || in tensorChunkAccessedByUnknownOp()
264 assert(read.transferReadOp && write.transferWriteOp && in hoistReadWrite()
271 << "\nHoist: " << *read.transferReadOp.getOperation() in hoistReadWrite()
280 forOp.moveOutOfLoop(read.transferReadOp); in hoistReadWrite()
291 read.transferReadOp.getSourceMutable().assign( in hoistReadWrite()
307 OpBuilder b(read.transferReadOp); in hoistReadWrite()
313 b, forOp, read.transferReadOp.getVector(), yieldFn); in hoistReadWrite()
377 if (!matchingRead.transferReadOp || in hoistRedundantVectorTransfersOnTensor()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 344 auto transferReadOp = in matchAndRewrite() local
346 if (!transferReadOp) in matchAndRewrite()
350 if (transferReadOp.getTransferRank() == 0) in matchAndRewrite()
353 if (transferReadOp.getMask() || transferReadOp.hasOutOfBoundsDim()) in matchAndRewrite()
363 permutationMap.compose(transferReadOp.getPermutationMap()); in matchAndRewrite()
365 op, op.getType(), transferReadOp.getSource(), in matchAndRewrite()
366 transferReadOp.getIndices(), AffineMapAttr::get(newMap), in matchAndRewrite()
367 transferReadOp.getPadding(), transferReadOp.getMask(), in matchAndRewrite()
368 transferReadOp.getInBoundsAttr()); in matchAndRewrite()
849 .Case([&](vector::TransferReadOp transferReadOp) { in convertVectorToNVVMCompatibleMMASync() argument
[all …]
|