Home
last modified time | relevance | path

Searched refs:segmentSizes (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp415 SmallVector<int32_t, 8> segmentSizes(8, 1); in build() local
416 segmentSizes.front() = asyncDependencies.size(); in build()
417 segmentSizes.back() = dynamicSharedMemorySize ? 1 : 0; in build()
419 builder.getI32VectorAttr(segmentSizes)); in build()
635 SmallVector<int32_t, 8> segmentSizes(8, 1); in parse() local
636 segmentSizes.front() = asyncDependencies.size(); in parse()
637 segmentSizes.back() = hasDynamicSharedMemorySize ? 1 : 0; in parse()
707 SmallVector<int32_t, 9> segmentSizes(9, 1); in build() local
708 segmentSizes.front() = asyncDependencies.size(); in build()
709 segmentSizes[segmentSizes.size() - 2] = dynamicSharedMemorySize ? 1 : 0; in build()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DValueRange.h57 OperandRangeRange split(ElementsAttr segmentSizes) const;
162 MutableOperandRangeRange split(NamedAttribute segmentSizes) const;
/llvm-project-15.0.7/mlir/lib/IR/
H A DOperationSupport.cpp380 OperandRangeRange OperandRange::split(ElementsAttr segmentSizes) const { in split()
381 return OperandRangeRange(*this, segmentSizes); in split()
482 MutableOperandRange::split(NamedAttribute segmentSizes) const { in split()
483 return MutableOperandRangeRange(*this, segmentSizes); in split()