Files
mlx/mlx/backend/cuda/reduce/all_reduce.cu