| /llvm-project/mlir/lib/Dialect/NVGPU/Transforms/ |
| H A D | CreateAsyncGroups.cpp | 158 op->walk([&](Operation *writeOp) { in createAsyncGroups() argument
208 Operation *writeOp = *copyToSharedMem.begin(); in createAsyncGroups() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 76 auto writeOp = cast<vector::TransferWriteOp>(op); in bufferizesToMemoryRead() local
107 auto writeOp = cast<vector::TransferWriteOp>(op); in bufferize() local
|
| H A D | VectorDistribute.cpp | 409 cloneWriteOp(RewriterBase & rewriter,WarpExecuteOnLane0Op warpOp,vector::TransferWriteOp writeOp,VectorType targetType,VectorType maybeMaskType) cloneWriteOp() argument
630 auto writeOp = dyn_cast_or_null<vector::TransferWriteOp>(lastNode); matchAndRewrite() local
[all...] |
| H A D | VectorTransforms.cpp | 1367 matchAndRewrite(vector::TransferWriteOp writeOp,PatternRewriter & rewriter) const matchAndRewrite() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | FoldTensorSubsetOps.cpp | 128 auto writeOp = getTensorOperand(insertSliceOp) matchAndRewrite() local
[all...] |
| /llvm-project/mlir/lib/Dialect/NVGPU/Utils/ |
| H A D | MMAUtils.cpp | 62 if (vector::TransferWriteOp writeOp = dyn_cast<vector::TransferWriteOp>(op)) { in getWarpMatrixInfo() local
|
| /llvm-project/mlir/lib/Dialect/Vector/Utils/ |
| H A D | VectorUtils.cpp | 317 [&](auto writeOp) { return writeOp.getOperand(1); }); in getMixedSizesXfer() argument
|
| /llvm-project/mlir/test/lib/Dialect/Vector/ |
| H A D | TestVectorTransforms.cpp | 102 auto insert = writeOp.getVector().getDefiningOp<InsertStridedSliceOp>(); in getShape() local
275 numLoops = writeOp.getVectorType().getRank(); in runOnOperation() local
|
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.cpp | 151 } else if (auto writeOp = dyn_cast<AffineWriteOpInterface>(op)) { in isLoopMemoryParallel() local
|
| /llvm-project/llvm/lib/ObjectYAML/ |
| H A D | MachOEmitter.cpp | 554 (this->*writeOp.second)(OS); in writeLinkEditData() local
|
| /llvm-project/mlir/lib/Dialect/GPU/TransformOps/ |
| H A D | GPUTransformOps.cpp | 183 if (auto writeOp = dyn_cast<vector::TransferWriteOp>(op)) { in getSubgroupMmaNativeVectorSize() local
|
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldMemRefAliasOps.cpp | 420 vector::TransferWriteOp writeOp, in preconditionsFoldSubViewOp() argument
|
| /llvm-project/mlir/lib/Bytecode/Writer/ |
| H A D | BytecodeWriter.cpp | 947 LogicalResult BytecodeWriter::writeOp(EncodingEmitter &emitter, Operation *op) { writeOp() function in BytecodeWriter
|
| /llvm-project/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 170 transferWriteSupportsMMAMatrixType(vector::TransferWriteOp writeOp) { in transferWriteSupportsMMAMatrixType() argument
|
| /llvm-project/mlir/lib/Target/LLVMIR/Dialect/OpenMP/ |
| H A D | OpenMPToLLVMIRTranslation.cpp | 1612 auto writeOp = cast<omp::AtomicWriteOp>(opInst); convertOmpAtomicWrite() local
|
| /llvm-project/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 4638 matchAndRewrite(TransferWriteOp writeOp,PatternRewriter & rewriter) const matchAndRewrite() argument
|