Fix CUDA all-reduce planning for large inputs#3603

sofinvalery wants to merge 1 commit into

ml-explore:mainml-explore/mlx:mainfrom

sofinvalery:fix-all-reducesofinvalery/mlx:fix-all-reduceCopy head branch name to clipboard