Home
last modified time | relevance | path

Searched refs:memorySpace (Results 1 – 18 of 18) sorted by relevance

/llvm-project-15.0.7/mlir/lib/IR/
H A DBuiltinTypes.cpp498 if (!memorySpace) in isSupportedMemorySpace()
514 if (memorySpace == 0) in wrapIntegerMemorySpace()
525 return memorySpace; in skipDefaultMemorySpace()
529 if (!memorySpace) in getMemorySpaceAsInt()
540 memorySpace = in setMemorySpace()
558 memorySpace = skipDefaultMemorySpace(memorySpace); in get()
574 memorySpace = skipDefaultMemorySpace(memorySpace); in getChecked()
592 memorySpace = skipDefaultMemorySpace(memorySpace); in get()
612 memorySpace = skipDefaultMemorySpace(memorySpace); in getChecked()
630 Attribute memorySpace = in get() local
[all …]
H A DTypeDetail.h135 bool isSupportedMemorySpace(Attribute memorySpace);
138 Attribute wrapIntegerMemorySpace(unsigned memorySpace, MLIRContext *ctx);
141 Attribute skipDefaultMemorySpace(Attribute memorySpace);
145 unsigned getMemorySpaceAsInt(Attribute memorySpace);
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp261 defaultUnknownTypeConverter(Value value, unsigned memorySpace, in defaultUnknownTypeConverter() argument
264 memorySpace); in defaultUnknownTypeConverter()
589 Optional<unsigned> memorySpace = None; in getBufferType() local
597 memorySpace = *queriedMemorySpace; in getBufferType()
604 if (!memorySpace.has_value()) in getBufferType()
605 memorySpace = options.defaultMemorySpace; in getBufferType()
608 if (!memorySpace.has_value()) in getBufferType()
703 IntegerType::get(tensorType.getContext(), 64), memorySpace); in getMemRefType()
729 memorySpace); in getMemRefTypeWithFullyDynamicLayout()
734 IntegerType::get(tensorType.getContext(), 64), memorySpace); in getMemRefTypeWithFullyDynamicLayout()
[all …]
H A DBufferizationOps.cpp174 unsigned memorySpace; in bufferize() local
176 memorySpace = *getMemorySpace(); in bufferize()
178 memorySpace = in bufferize()
181 memorySpace = *options.defaultMemorySpace; in bufferize()
189 AffineMap(), memorySpace); in bufferize()
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DBuiltinTypes.cpp237 MlirAttribute memorySpace) { in mlirMemRefTypeGet() argument
243 unwrap(memorySpace))); in mlirMemRefTypeGet()
249 MlirAttribute memorySpace) { in mlirMemRefTypeGetChecked() argument
256 unwrap(memorySpace))); in mlirMemRefTypeGetChecked()
261 MlirAttribute memorySpace) { in mlirMemRefTypeContiguousGet() argument
264 MemRefLayoutAttrInterface(), unwrap(memorySpace))); in mlirMemRefTypeContiguousGet()
270 MlirAttribute memorySpace) { in mlirMemRefTypeContiguousGetChecked() argument
293 MlirAttribute memorySpace) { in mlirUnrankedMemRefTypeGet() argument
295 UnrankedMemRefType::get(unwrap(elementType), unwrap(memorySpace))); in mlirUnrankedMemRefTypeGet()
300 MlirAttribute memorySpace) { in mlirUnrankedMemRefTypeGetChecked() argument
[all …]
/llvm-project-15.0.7/mlir/include/mlir-c/
H A DBuiltinTypes.h245 MlirAttribute memorySpace);
251 MlirAttribute layout, MlirAttribute memorySpace);
259 const int64_t *shape, MlirAttribute memorySpace);
265 MlirAttribute memorySpace);
270 mlirUnrankedMemRefTypeGet(MlirType elementType, MlirAttribute memorySpace);
275 MlirLocation loc, MlirType elementType, MlirAttribute memorySpace);
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DTypeParser.cpp225 Attribute memorySpace; in parseMemRefType() local
245 } else if (memorySpace) { in parseMemRefType()
248 memorySpace = attr; in parseMemRefType()
255 if (memorySpace) in parseMemRefType()
272 return getChecked<UnrankedMemRefType>(loc, elementType, memorySpace); in parseMemRefType()
275 memorySpace); in parseMemRefType()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Pipelines/
H A DSparseTensorPipelines.cpp35 options.unknownTypeConverterFn = [](Value value, unsigned memorySpace, in getBufferizationOptions()
38 value.getType().cast<TensorType>(), memorySpace); in getBufferizationOptions()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferize.cpp199 opt.unknownTypeConverterFn = [=](Value value, unsigned memorySpace, in runOnOperation()
205 tensorType, memorySpace); in runOnOperation()
211 memorySpace); in runOnOperation()
490 options.unknownTypeConverterFn = [](Value value, unsigned memorySpace, in getPartialBufferizationOptions()
493 value.getType().cast<TensorType>(), memorySpace); in getPartialBufferizationOptions()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.h170 layout(other.getLayout()), memorySpace(other.getMemorySpace()) {} in Builder()
192 memorySpace = newMemorySpace; in setMemorySpace()
200 return MemRefType::get(shape, elementType, layout, memorySpace); in MemRefType()
207 Attribute memorySpace; variable
H A DBuiltinTypes.td531 "Attribute":$memorySpace
537 CArg<"Attribute", "{}">:$memorySpace)>,
541 CArg<"Attribute", "{}">:$memorySpace)>,
840 let parameters = (ins "Type":$elementType, "Attribute":$memorySpace);
844 "Attribute":$memorySpace), [{
846 Attribute nonDefaultMemorySpace = skipDefaultMemorySpace(memorySpace);
851 "unsigned":$memorySpace), [{
854 wrapIntegerMemorySpace(memorySpace, elementType.getContext());
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DIRTypes.cpp441 PyAttribute *layout, PyAttribute *memorySpace, in bindDerived()
445 memorySpace ? *memorySpace : mlirAttributeGetNull(); in bindDerived()
500 [](PyType &elementType, PyAttribute *memorySpace, in bindDerived()
503 if (memorySpace) in bindDerived()
504 memSpaceAttr = *memorySpace; in bindDerived()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.h534 unsigned memorySpace = 0);
539 unsigned memorySpace = 0);
544 unsigned memorySpace = 0);
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DUtils.h370 int memorySpace = -1);
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp1289 int memorySpace) { in getMemoryFootprintBytes() argument
1331 int memorySpace) { in getMemoryFootprintBytes() argument
1335 std::next(Block::iterator(forInst)), memorySpace); in getMemoryFootprintBytes()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp984 unsigned memorySpace) { in verifyAttributions() argument
990 if (type.getMemorySpaceAsInt() != memorySpace) { in verifyAttributions()
992 << "expected memory space " << memorySpace << " in attribution"; in verifyAttributions()
/llvm-project-15.0.7/mlir/test/Dialect/GPU/
H A Dinvalid.mlir504 …// expected-error @+1 {{source memorySpace kGenericMemorySpace, kSharedMemorySpace or kGlobalMemor…
528 …// expected-error @+1 {{destination memorySpace of kGenericMemorySpace, kGlobalMemorySpace or kSha…
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1013 unsigned memorySpace = in extractPointersAndOffset() local
1018 LLVM::LLVMPointerType::get(llvmElementType, memorySpace)); in extractPointersAndOffset()