Lines Matching refs:numThreadDims
135 unsigned numThreadDims) { in checkAffineLoopNestMappable() argument
136 if (numBlockDims < 1 || numThreadDims < 1) { in checkAffineLoopNestMappable()
144 if (numThreadDims > 3) { in checkAffineLoopNestMappable()
147 return checkAffineLoopNestMappableImpl(forOp, numBlockDims + numThreadDims); in checkAffineLoopNestMappable()
157 unsigned numBlockDims, unsigned numThreadDims);
221 unsigned numThreadDims) { in createLaunch() argument
226 (numBlockDims < 3 || numThreadDims < 3) in createLaunch()
232 Value blockSizeX = numThreadDims > 0 ? dims[numBlockDims] : constOne; in createLaunch()
233 Value blockSizeY = numThreadDims > 1 ? dims[numBlockDims + 1] : constOne; in createLaunch()
234 Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne; in createLaunch()
285 unsigned numThreadDims) { in convertAffineLoopNestToGPULaunch() argument
286 if (failed(checkAffineLoopNestMappable(forOp, numBlockDims, numThreadDims))) in convertAffineLoopNestToGPULaunch()
291 converter.collectBounds(forOp, numBlockDims + numThreadDims); in convertAffineLoopNestToGPULaunch()
294 converter.createLaunch(forOp, *maybeInnerLoop, numBlockDims, numThreadDims); in convertAffineLoopNestToGPULaunch()
301 unsigned numThreadDims) { in convertAffineLoopNestToGPULaunch() argument
302 return ::convertAffineLoopNestToGPULaunch(forOp, numBlockDims, numThreadDims); in convertAffineLoopNestToGPULaunch()