Searched refs:multimem_one_shot_all_reduce_kernel (Results 1 – 1 of 1) sorted by relevance
170 static __global__ void multimem_one_shot_all_reduce_kernel( in multimem_one_shot_all_reduce_kernel() function225 multimem_one_shot_all_reduce_kernel<scalar_t, kernel_alignment> \ in multimem_one_shot_all_reduce()