| /llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 160 if (trueType.getMemorySpaceAsInt() != falseType.getMemorySpaceAsInt()) in bufferize()
175 stridedLayout, trueType.getMemorySpaceAsInt()); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | BuiltinTypes.cpp | 419 unsigned BaseMemRefType::getMemorySpaceAsInt() const { in getMemorySpaceAsInt() function in BaseMemRefType
421 return rankedMemRefTy.getMemorySpaceAsInt(); in getMemorySpaceAsInt()
422 return cast<UnrankedMemRefType>().getMemorySpaceAsInt(); in getMemorySpaceAsInt()
528 unsigned mlir::detail::getMemorySpaceAsInt(Attribute memorySpace) { in getMemorySpaceAsInt() function in mlir::detail
545 unsigned MemRefType::getMemorySpaceAsInt() const { in getMemorySpaceAsInt() function in MemRefType
546 return detail::getMemorySpaceAsInt(getMemorySpace()); in getMemorySpaceAsInt()
702 unsigned UnrankedMemRefType::getMemorySpaceAsInt() const { in getMemorySpaceAsInt() function in UnrankedMemRefType
703 return detail::getMemorySpaceAsInt(getMemorySpace()); in getMemorySpaceAsInt()
|
| H A D | TypeDetail.h | 145 unsigned getMemorySpaceAsInt(Attribute memorySpace);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/IR/ |
| H A D | NVGPUDialect.cpp | 76 if (dstMemref.getMemorySpaceAsInt() != workgroupAddressSpace) in verify()
225 if (!(srcMemref.getMemorySpaceAsInt() == smemAddressSpace)) in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/AMDGPU/IR/ |
| H A D | AMDGPUDialect.cpp | 35 if (bufferType.getMemorySpaceAsInt() != 0) in verifyRawBufferOp()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/ |
| H A D | TypeConverter.cpp | 308 LLVM::LLVMPointerType::get(elementType, type.getMemorySpaceAsInt()); in getMemRefDescriptorFields()
326 unsigned space = type.getMemorySpaceAsInt(); in getMemRefDescriptorSize()
359 unsigned space = type.getMemorySpaceAsInt(); in getUnrankedMemRefDescriptorSize()
402 return LLVM::LLVMPointerType::get(elementType, type.getMemorySpaceAsInt()); in convertMemRefToBarePtr()
|
| H A D | Pattern.cpp | 117 type.getMemorySpaceAsInt()); in getElementPtrType()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/ |
| H A D | OptimizeSharedMemory.cpp | 176 if (!memRefType || memRefType.getMemorySpaceAsInt() != in optimizeSharedMemoryReadsAndWrites()
256 .getMemorySpaceAsInt() != in runOnOperation()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.h | 131 return getSrcMemRef().getType().cast<MemRefType>().getMemorySpaceAsInt(); in getSrcMemorySpace()
152 return getDstMemRef().getType().cast<MemRefType>().getMemorySpaceAsInt(); in getDstMemorySpace()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 34 if (srcType.getMemorySpaceAsInt() != destType.getMemorySpaceAsInt()) in castOrReallocMemRefValue()
179 copyBuffer.getType().cast<BaseMemRefType>().getMemorySpaceAsInt(); in bufferize()
|
| H A D | BufferizableOpInterface.cpp | 119 copyBufferType->getMemorySpaceAsInt())); in allocateTensorForShapedValue()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToSPIRV/ |
| H A D | MemRefToSPIRV.cpp | 94 spirv::StorageClass::Workgroup) != type.getMemorySpaceAsInt()) in isAllocationSupported()
98 spirv::StorageClass::Function) != type.getMemorySpaceAsInt()) in isAllocationSupported()
121 type.getMemorySpaceAsInt()); in getAtomicOpScope()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 71 sourceMemRefType.getMemorySpaceAsInt()); in bufferize()
141 bufferType.getMemorySpaceAsInt()); in bufferize()
165 AffineMap(), bufferType.getMemorySpaceAsInt()); in bufferize()
810 srcBuffer->getType().cast<BaseMemRefType>().getMemorySpaceAsInt()); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 376 unsigned addressSpace = unrankedMemRefType.getMemorySpaceAsInt(); in extractSizeOfUnrankedMemRef()
627 initialValue, alignment, type.getMemorySpaceAsInt()); in matchAndRewrite()
655 unsigned memSpace = type.getMemorySpaceAsInt(); in allocateBuffer()
1014 operandType.cast<UnrankedMemRefType>().getMemorySpaceAsInt(); in extractPointersAndOffset()
1214 unsigned addressSpace = targetType.getMemorySpaceAsInt(); in convertSourceMemRefToDescriptor()
1685 viewMemRefType.getMemorySpaceAsInt()), in matchAndRewrite()
1694 viewMemRefType.getMemorySpaceAsInt()), in matchAndRewrite()
1928 srcMemRefType.getMemorySpaceAsInt()), in matchAndRewrite()
1939 srcMemRefType.getMemorySpaceAsInt()), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | BufferResultsToOutParams.cpp | 138 AffineMap(), memrefType.getMemorySpaceAsInt()); in updateCalls()
|
| H A D | FuncBufferizableOpInterfaceImpl.cpp | 89 layoutAttr.getValue(), rankedMemrefType.getMemorySpaceAsInt()); in getBufferizedFunctionArgType()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 368 LLVM::LLVMPointerType::get(i8Ty, dstMemrefType.getMemorySpaceAsInt()); in matchAndRewrite()
376 LLVM::LLVMPointerType::get(i8Ty, srcMemrefType.getMemorySpaceAsInt()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | BufferizableOpInterfaceImpl.cpp | 186 if (thenMemrefType.getMemorySpaceAsInt() != in bufferize()
187 elseMemrefType.getMemorySpaceAsInt()) in bufferize()
192 thenMemrefType.getMemorySpaceAsInt()); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/ |
| H A D | LowerGpuOpsToNVVMOps.cpp | 188 if (type.getMemorySpaceAsInt() != in runOnOperation()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | BuiltinTypes.h | 142 unsigned getMemorySpaceAsInt() const;
|
| H A D | BuiltinTypes.td | 565 unsigned getMemorySpaceAsInt() const;
873 unsigned getMemorySpaceAsInt() const;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 990 if (type.getMemorySpaceAsInt() != memorySpace) { in verifyAttributions()
1172 auto srcMemSpace = srcMemrefType.getMemorySpaceAsInt(); in verify()
1200 auto dstMemSpace = dstMemrefType.getMemorySpaceAsInt(); in verify()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | SPIRVConversion.cpp | 381 type.getMemorySpaceAsInt()); in convertBoolMemrefType()
433 type.getMemorySpaceAsInt()); in convertMemrefType()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToLLVM/ |
| H A D | ConvertVectorToLLVM.cpp | 106 memRefType.getMemorySpaceAsInt() != 0) in getIndexedPtrs()
118 auto pType = LLVM::LLVMPointerType::get(vt, memRefType.getMemorySpaceAsInt()); in castDataPtr()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 1764 srcType.getMemorySpaceAsInt()); in computeExpandedType()
1928 srcType.getMemorySpaceAsInt()); in computeCollapsedType()
1972 *computedLayout, srcType.getMemorySpaceAsInt()); in verify()
|