Lines Matching defs:totalNumThreads

38                                                  int totalNumThreads,
61   status = inferNumThreads(totalNumThreads, copySizes, desiredVectorSize,
129 /// mapping a copy of size `sizes`, while being bounded by `totalNumThreads` and
181 transform::gpu::CopyMappingInfo::inferNumThreads(int64_t totalNumThreads,
197           inferNumThreadsImpl(totalNumThreads, sizes, localVectorSize);
209   return inferNumThreadsImpl(totalNumThreads, sizes, desiredVectorSize);
214     int64_t totalNumThreads, ArrayRef<int64_t> sizes,
219   LDBG("inferNumThreadsImpl with totalNumThreads: "
220        << totalNumThreads << " and vectorSize: " << desiredVectorSize);
227   if (scaledSizes.back() > totalNumThreads) {
232       maximizeNumThreads(scaledSizes, 0, totalNumThreads);
244   if (totalNumThreadsUsed == 0 || totalNumThreadsUsed > totalNumThreads) {
251   if (totalNumThreadsUsed == totalNumThreads)