Lines Matching refs:NumThreads
84 uint32_t NumThreads = GetNumberOfOmpThreads(isSPMDExecutionMode); in nvptx_parallel_reduce_nowait() local
85 if (NumThreads == 1) in nvptx_parallel_reduce_nowait()
99 uint32_t WarpsNeeded = (NumThreads + WARPSIZE - 1) / WARPSIZE; in nvptx_parallel_reduce_nowait()
106 if ((NumThreads % WARPSIZE == 0) || (WarpId < WarpsNeeded - 1)) in nvptx_parallel_reduce_nowait()
108 else if (NumThreads > 1) // Only SPMD execution mode comes thru this case. in nvptx_parallel_reduce_nowait()
110 /*LaneCount=*/NumThreads % WARPSIZE, in nvptx_parallel_reduce_nowait()
117 if (NumThreads > WARPSIZE) { in nvptx_parallel_reduce_nowait()
144 if (NumThreads > WARPSIZE) { in nvptx_parallel_reduce_nowait()
145 uint32_t WarpsNeeded = (NumThreads + WARPSIZE - 1) / WARPSIZE; in nvptx_parallel_reduce_nowait()
204 uint32_t NumThreads = in __kmpc_nvptx_teams_reduce_nowait_v2() local
270 NumThreads = roundToWarpsize(__kmpc_impl_min(NumThreads, NumRecs)); in __kmpc_nvptx_teams_reduce_nowait_v2()
271 if (ThreadId >= NumThreads) in __kmpc_nvptx_teams_reduce_nowait_v2()
276 for (uint32_t i = NumThreads + ThreadId; i < NumRecs; i += NumThreads) in __kmpc_nvptx_teams_reduce_nowait_v2()
280 if (NumThreads > 1) { in __kmpc_nvptx_teams_reduce_nowait_v2()
285 uint32_t ActiveThreads = __kmpc_impl_min(NumRecs, NumThreads); in __kmpc_nvptx_teams_reduce_nowait_v2()