Lines Matching refs:numThreads
197 OpFoldResult numThreads, in canOmitTileOffsetInBoundsCheck() argument
200 Optional<int64_t> numThreadsConst = getConstantIntValue(numThreads); in canOmitTileOffsetInBoundsCheck()
220 RewriterBase &b, TilingInterface op, ArrayRef<OpFoldResult> numThreads, in tileToForeachThreadOpImpl() argument
237 llvm::to_vector(llvm::make_filter_range(numThreads, [](OpFoldResult ofr) { in tileToForeachThreadOpImpl()
264 bool overflow = loopIdx >= numThreads.size(); in tileToForeachThreadOpImpl()
265 bool isZero = !overflow && isConstantIntValue(numThreads[loopIdx], 0); in tileToForeachThreadOpImpl()
339 ArrayRef<OpFoldResult> numThreads, in tileToForeachThreadOp() argument
341 return tileToForeachThreadOpImpl(b, op, numThreads, /*nominalTileSizes=*/None, in tileToForeachThreadOp()
352 SmallVector<OpFoldResult> numThreads; in tileToForeachThreadOpUsingTileSizes() local
353 numThreads.reserve(nLoops); in tileToForeachThreadOpUsingTileSizes()
362 numThreads.push_back(numTiles); in tileToForeachThreadOpUsingTileSizes()
364 return tileToForeachThreadOpImpl(b, op, numThreads, in tileToForeachThreadOpUsingTileSizes()