Searched refs:dstMemref (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/IR/ |
| H A D | NVGPUDialect.cpp | 70 auto dstMemref = getDst().getType().cast<MemRefType>(); in verify() local
74 if (!isLastMemrefDimUnitStride(dstMemref)) in verify()
76 if (dstMemref.getMemorySpaceAsInt() != workgroupAddressSpace) in verify()
79 if (dstMemref.getElementType() != srcMemref.getElementType()) in verify()
84 if (size_t(dstMemref.getRank()) != getDstIndices().size()) in verify()
85 return emitOpError() << "expected " << dstMemref.getRank() in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | MultiBuffer.cpp | 137 auto dstMemref = in multiBuffer() local
141 Value subview = builder.create<memref::SubViewOp>(loc, dstMemref, newAlloc, in multiBuffer()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 712 FailureOr<Value> dstMemref = in bufferize() local
714 if (failed(dstMemref)) in bufferize()
718 auto dstMemrefType = dstMemref->getType().cast<MemRefType>(); in bufferize()
725 loc, subviewMemRefType, *dstMemref, mixedOffsets, mixedSizes, in bufferize()
737 replaceOpWithBufferizedValues(rewriter, op, *dstMemref); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | WmmaOpsToNvvm.cpp | 169 loc, subgroupMmaStoreMatrixOp.dstMemref().getType().cast<MemRefType>(), in matchAndRewrite()
170 adaptor.dstMemref(), adaptor.indices(), rewriter); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 1134 Arg<MemRefOf<[F16, F32]>, "",[MemWrite]>:$dstMemref,
1139 $src`,` $dstMemref`[`$indices`]` attr-dict `:` type($src)`,` type($dstMemref)
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1197 auto dstType = dstMemref().getType(); in verify()
|