Searched refs:shflFct (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/openmp/libomptarget/deviceRTLs/common/src/ |
| D | reduction.cu | 36 kmp_ShuffleReductFctPtr shflFct) { in gpu_regular_warp_reduce() argument
38 shflFct(reduce_data, /*LaneId - not used= */ 0, in gpu_regular_warp_reduce()
44 kmp_ShuffleReductFctPtr shflFct, in gpu_irregular_warp_reduce() argument
51 shflFct(reduce_data, /*LaneId = */ tid, /*Offset=*/mask, /*AlgoVersion=*/1); in gpu_irregular_warp_reduce()
59 gpu_irregular_simd_reduce(void *reduce_data, kmp_ShuffleReductFctPtr shflFct) { in gpu_irregular_simd_reduce() argument
71 shflFct(reduce_data, /*LaneId =*/logical_lane_id, in gpu_irregular_simd_reduce()
81 kmp_ShuffleReductFctPtr shflFct, kmp_InterWarpCopyFctPtr cpyFct, in nvptx_parallel_reduce_nowait() argument
107 gpu_regular_warp_reduce(reduce_data, shflFct); in nvptx_parallel_reduce_nowait()
109 gpu_irregular_warp_reduce(reduce_data, shflFct, in nvptx_parallel_reduce_nowait()
123 gpu_irregular_warp_reduce(reduce_data, shflFct, WarpsNeeded, in nvptx_parallel_reduce_nowait()
[all …]
|
| /external/llvm-project/openmp/libomptarget/deviceRTLs/ |
| D | interface.h | 349 void *reduce_data, kmp_ShuffleReductFctPtr shflFct,
353 int32_t num_of_records, void *reduce_data, kmp_ShuffleReductFctPtr shflFct,
|