Lines Matching defs:numThreadDims

138                                                  unsigned numThreadDims) {
139   if (numBlockDims < 1 || numThreadDims < 1) {
147   if (numThreadDims > 3) {
150   return checkAffineLoopNestMappableImpl(forOp, numBlockDims + numThreadDims);
161                     unsigned numBlockDims, unsigned numThreadDims);
214 // and to have (numBlockDims + numThreadDims) perfectly nested loops between
219                                             unsigned numThreadDims) {
224       (numBlockDims < 3 || numThreadDims < 3)
230   Value blockSizeX = numThreadDims > 0 ? dims[numBlockDims] : constOne;
231   Value blockSizeY = numThreadDims > 1 ? dims[numBlockDims + 1] : constOne;
232   Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne;
283                                                       unsigned numThreadDims) {
284   if (failed(checkAffineLoopNestMappable(forOp, numBlockDims, numThreadDims)))
289       converter.collectBounds(forOp, numBlockDims + numThreadDims);
292   converter.createLaunch(forOp, *maybeInnerLoop, numBlockDims, numThreadDims);
299                                                      unsigned numThreadDims) {
300   return ::convertAffineLoopNestToGPULaunch(forOp, numBlockDims, numThreadDims);