Searched refs:bufferType (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/AMDGPU/IR/ |
| H A D | AMDGPUDialect.cpp | 34 MemRefType bufferType = op.getMemref().getType().template cast<MemRefType>(); in verifyRawBufferOp() local
35 if (bufferType.getMemorySpaceAsInt() != 0) in verifyRawBufferOp()
38 if (!bufferType.hasRank()) in verifyRawBufferOp()
41 if (static_cast<int64_t>(op.getIndices().size()) != bufferType.getRank()) in verifyRawBufferOp()
42 return op.emitOpError("Expected " + Twine(bufferType.getRank()) + in verifyRawBufferOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | MemoryPromotion.cpp | 151 auto bufferType = MemRefType::get(type.getShape(), type.getElementType(), {}, in promoteToWorkgroupMemory() local
153 Value attribution = op.addWorkgroupAttribution(bufferType, value.getLoc()); in promoteToWorkgroupMemory()
|
| H A D | AllReduceLowering.cpp | 163 auto bufferType = MemRefType::get({kSubgroupSize}, valueType, AffineMap{}, in createWorkgroupBuffer() local
165 return funcOp.addWorkgroupAttribution(bufferType, rewriter.getUnknownLoc()); in createWorkgroupBuffer()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 119 auto bufferType = buffer.getType().cast<MemRefType>(); in bufferize() local
125 if (bufferType.getLayout().isIdentity()) { in bufferize()
129 layout, bufferType.getMemorySpace()); in bufferize()
135 if (failed(getStridesAndOffset(bufferType, strides, offset))) in bufferize()
141 bufferType.getMemorySpaceAsInt()); in bufferize()
153 bufferType, collapseShapeOp.getReassociationIndices()); in bufferize()
165 AffineMap(), bufferType.getMemorySpaceAsInt()); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToSCF/ |
| H A D | VectorToSCF.cpp | 290 auto bufferType = MemRefType::get({}, xferOp.getVectorType()); in allocBuffers() local
291 result.dataBuffer = b.create<memref::AllocaOp>(loc, bufferType); in allocBuffers()
403 auto bufferType = buffer.getType().dyn_cast<ShapedType>(); in rewriteOp() local
404 auto vecType = bufferType.getElementType().dyn_cast<VectorType>(); in rewriteOp()
427 auto bufferType = buffer.getType().dyn_cast<ShapedType>(); in handleOutOfBoundsDim() local
428 auto vecType = bufferType.getElementType().dyn_cast<VectorType>(); in handleOutOfBoundsDim()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Transforms/ |
| H A D | AbstractResult.cpp | 216 auto bufferType = fir::ReferenceType::get(resultType); in runOnOperation() local
218 newArg = rewriter.create<fir::BoxAddrOp>(loc, bufferType, newArg); in runOnOperation()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Builder/ |
| H A D | Character.cpp | 260 auto bufferType = buffer.getType(); in createElementAddr() local
261 assert(fir::isa_ref_type(bufferType)); in createElementAddr()
262 assert(isCharacterScalar(bufferType)); in createElementAddr()
263 auto charTy = recoverCharacterType(bufferType); in createElementAddr()
|