Home
last modified time | relevance | path

Searched refs:blockDims (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DInferIntRangeInterfaceImpls.cpp92 KernelDim3 blockDims = getBlockSize(); in inferResultRanges() local
94 setRange(argRanges[3], blockDims.x, threadIds.x); in inferResultRanges()
95 setRange(argRanges[4], blockDims.y, threadIds.y); in inferResultRanges()
96 setRange(argRanges[5], blockDims.z, threadIds.z); in inferResultRanges()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DMemoryPromotion.cpp61 SmallVector<Value, 3> threadIds, blockDims; in insertCopyLoops() local
64 blockDims.push_back(b.create<gpu::BlockDimOp>(indexType, dim)); in insertCopyLoops()
85 {blockDims[en.index()]}); in insertCopyLoops()