Home
last modified time | relevance | path

Searched defs:memorySpace (Results 1 – 23 of 23) sorted by relevance

/llvm-project/mlir/lib/IR/
H A DBuiltinTypes.cpp478 isSupportedMemorySpace(Attribute memorySpace) isSupportedMemorySpace() argument
494 wrapIntegerMemorySpace(unsigned memorySpace,MLIRContext * ctx) wrapIntegerMemorySpace() argument
502 skipDefaultMemorySpace(Attribute memorySpace) skipDefaultMemorySpace() argument
510 getMemorySpaceAsInt(Attribute memorySpace) getMemorySpaceAsInt() argument
526 get(ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) get() argument
541 getChecked(function_ref<InFlightDiagnostic ()> emitErrorFn,ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) getChecked() argument
556 get(ArrayRef<int64_t> shape,Type elementType,AffineMap map,Attribute memorySpace) get() argument
576 getChecked(function_ref<InFlightDiagnostic ()> emitErrorFn,ArrayRef<int64_t> shape,Type elementType,AffineMap map,Attribute memorySpace) getChecked() argument
605 Attribute memorySpace = get() local
626 Attribute memorySpace = getChecked() local
636 verify(function_ref<InFlightDiagnostic ()> emitError,ArrayRef<int64_t> shape,Type elementType,MemRefLayoutAttrInterface layout,Attribute memorySpace) verify() argument
665 verify(function_ref<InFlightDiagnostic ()> emitError,Type elementType,Attribute memorySpace) verify() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DConvertToDestinationStyle.cpp194 createAllocationForTensor(RewriterBase & rewriter,Location loc,Value value,const linalg::BufferizeToAllocationOptions & options,Attribute memorySpace={}) createAllocationForTensor() argument
224 bufferizeToAllocation(RewriterBase & rewriter,const linalg::BufferizeToAllocationOptions & options,PadOp padOp,Attribute memorySpace,Operation * insertionPoint) bufferizeToAllocation() argument
263 bufferizeToAllocation(RewriterBase & rewriter,const linalg::BufferizeToAllocationOptions & options,vector::MaskOp maskOp,Attribute memorySpace,Operation * insertionPoint) bufferizeToAllocation() argument
325 bufferizeToAllocation(RewriterBase & rewriter,const linalg::BufferizeToAllocationOptions & options,bufferization::AllocTensorOp allocTensorOp,Attribute memorySpace,Operation * insertionPoint) bufferizeToAllocation() argument
473 bufferizeToAllocation(RewriterBase & rewriter,const linalg::BufferizeToAllocationOptions & options,Operation * op,Attribute memorySpace,Operation * insertionPoint) bufferizeToAllocation() argument
[all...]
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp196 std::optional<Attribute> memorySpace = copyBufferType->getMemorySpace(); in allocateTensorForShapedValue() local
316 defaultFunctionArgTypeConverter(TensorType type, Attribute memorySpace, in defaultFunctionArgTypeConverter() argument
323 defaultUnknownTypeConverter(Value value, Attribute memorySpace, in defaultUnknownTypeConverter() argument
365 __anon17a846d30502(TensorType tensorType, Attribute memorySpace, func::FuncOp funcOp, const BufferizationOptions &options) setFunctionBoundaryTypeConversion() argument
767 getMemRefType(Value value,const BufferizationOptions & options,MemRefLayoutAttrInterface layout,Attribute memorySpace) getMemRefType() argument
791 getMemRefTypeWithFullyDynamicLayout(TensorType tensorType,Attribute memorySpace) getMemRefTypeWithFullyDynamicLayout() argument
815 getMemRefTypeWithStaticIdentityLayout(TensorType tensorType,Attribute memorySpace) getMemRefTypeWithStaticIdentityLayout() argument
[all...]
H A DBufferizationOps.cpp231 Attribute memorySpace; in getBufferType() local
277 build(OpBuilder & builder,OperationState & result,TensorType type,ValueRange dynamicSizes,Value copy,IntegerAttr memorySpace) build() argument
/llvm-project/mlir/lib/CAPI/IR/
H A DBuiltinTypes.cpp389 mlirMemRefTypeGet(MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute layout,MlirAttribute memorySpace) mlirMemRefTypeGet() argument
401 mlirMemRefTypeGetChecked(MlirLocation loc,MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute layout,MlirAttribute memorySpace) mlirMemRefTypeGetChecked() argument
413 mlirMemRefTypeContiguousGet(MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute memorySpace) mlirMemRefTypeContiguousGet() argument
422 mlirMemRefTypeContiguousGetChecked(MlirLocation loc,MlirType elementType,intptr_t rank,const int64_t * shape,MlirAttribute memorySpace) mlirMemRefTypeContiguousGetChecked() argument
461 mlirUnrankedMemRefTypeGet(MlirType elementType,MlirAttribute memorySpace) mlirUnrankedMemRefTypeGet() argument
468 mlirUnrankedMemRefTypeGetChecked(MlirLocation loc,MlirType elementType,MlirAttribute memorySpace) mlirUnrankedMemRefTypeGetChecked() argument
[all...]
/llvm-project/mlir/lib/Conversion/MemRefToSPIRV/
H A DMapMemRefStorageClassPass.cpp75 unsigned memorySpace = intAttr.getInt(); in mapMemorySpaceToVulkanStorageClass() local
129 unsigned memorySpace = intAttr.getInt(); in mapMemorySpaceToOpenCLStorageClass() local
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferUtils.cpp105 getGlobalFor(arith::ConstantOp constantOp,uint64_t alignment,Attribute memorySpace) getGlobalFor() argument
H A DBufferize.cpp242 __anon99eb86680a02(Value value, Attribute memorySpace, const BufferizationOptions &options) runOnOperation() argument
629 __anon99eb86681002(Value value, Attribute memorySpace, const BufferizationOptions &options) getPartialBufferizationOptions() argument
/llvm-project/mlir/lib/Dialect/NVGPU/IR/
H A DNVGPUDialect.cpp50 bool nvgpu::NVGPUDialect::isSharedMemoryAddressSpace(Attribute memorySpace) { in isSharedMemoryAddressSpace() argument
61 Attribute memorySpace = type.getMemorySpace(); in hasSharedMemoryAddressSpace() local
/llvm-project/mlir/lib/Dialect/Arith/Transforms/
H A DBufferizableOpInterfaceImpl.cpp35 Attribute memorySpace; in bufferize() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsificationAndBufferizationPass.cpp217 __anon3c3b2a380202(Value value, Attribute memorySpace, const BufferizationOptions &options) getBufferizationOptionsForSparsification() argument
/llvm-project/mlir/lib/AsmParser/
H A DTypeParser.cpp211 Attribute memorySpace; parseMemRefType() local
/llvm-project/mlir/lib/Dialect/AMDGPU/IR/
H A DAMDGPUDialect.cpp67 Attribute memorySpace = bufferType.getMemorySpace(); verifyRawBufferOp() local
/llvm-project/mlir/include/mlir/IR/
H A DBuiltinTypes.h240 Attribute memorySpace; global() variable
/llvm-project/mlir/lib/Bindings/Python/
H A DIRTypes.cpp619 __anon2092a0da2902(std::vector<int64_t> shape, PyType &elementType, PyAttribute *layout, PyAttribute *memorySpace, DefaultingPyLocation loc) bindDerived() argument
685 __anon2092a0da2e02(PyType &elementType, PyAttribute *memorySpace, DefaultingPyLocation loc) bindDerived() argument
/llvm-project/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp171 isWorkgroupMemoryAddressSpace(Attribute memorySpace) isWorkgroupMemoryAddressSpace() argument
180 Attribute memorySpace = type.getMemorySpace(); hasWorkgroupMemoryAddressSpace() local
485 verifyAttributions(Operation * op,ArrayRef<BlockArgument> attributions,gpu::AddressSpace memorySpace) verifyAttributions() argument
/llvm-project/mlir/lib/Conversion/GPUCommon/
H A DGPUOpsLowering.cpp718 gpu::AddressSpace memorySpace = memorySpaceAttr.getValue(); populateGpuMemorySpaceAttributeConversions() local
/llvm-project/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp1912 int memorySpace) { in getMemoryFootprintBytes() argument
1954 int memorySpace) { in getMemoryFootprintBytes() argument
/llvm-project/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp230 Attribute memorySpace = {}; getMbarrierMemorySpace() local
243 Attribute memorySpace = nvgpu::getMbarrierMemorySpace(context, barrierType); getMBarrierMemrefType() local
/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h391 std::optional<Attribute> memorySpace; member
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp59 Attribute memorySpace = maybeSrcBufferType->getMemorySpace(); getBufferType() local
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp257 build(OpBuilder & b,OperationState & result,Value target,Attribute memorySpace) build() argument
270 build(OpBuilder & b,OperationState & result,Value target,int64_t memorySpace) build() argument
346 Attribute memorySpace = apply() local
/llvm-project/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1011 unsigned memorySpace = *typeConverter.getMemRefAddressSpace( extractPointersAndOffset() local