Lines Matching refs:numBlockDims
134 unsigned numBlockDims, in checkAffineLoopNestMappable() argument
136 if (numBlockDims < 1 || numThreadDims < 1) { in checkAffineLoopNestMappable()
141 if (numBlockDims > 3) { in checkAffineLoopNestMappable()
147 return checkAffineLoopNestMappableImpl(forOp, numBlockDims + numThreadDims); in checkAffineLoopNestMappable()
157 unsigned numBlockDims, unsigned numThreadDims);
220 unsigned numBlockDims, in createLaunch() argument
226 (numBlockDims < 3 || numThreadDims < 3) in createLaunch()
229 Value gridSizeX = numBlockDims > 0 ? dims[0] : constOne; in createLaunch()
230 Value gridSizeY = numBlockDims > 1 ? dims[1] : constOne; in createLaunch()
231 Value gridSizeZ = numBlockDims > 2 ? dims[2] : constOne; 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()
264 en.index() < numBlockDims in createLaunch()
266 : getDim3Value(launchOp.getThreadIds(), en.index() - numBlockDims); in createLaunch()
284 unsigned numBlockDims, 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()
300 unsigned numBlockDims, in convertAffineLoopNestToGPULaunch() argument
302 return ::convertAffineLoopNestToGPULaunch(forOp, numBlockDims, numThreadDims); in convertAffineLoopNestToGPULaunch()