Searched refs:multimem_all_reduce_kernel (Results 1 – 1 of 1) sorted by relevance
78 static __global__ void multimem_all_reduce_kernel( in multimem_all_reduce_kernel() function141 multimem_all_reduce_kernel<scalar_t, kernel_alignment> \ in multimem_all_reduce_()