zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 16:48:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix allocation bug in NCCL (#2530)

Browse Source
This commit is contained in:
Awni Hannun
2025-08-22 14:39:43 -07:00
committed by GitHub
parent 068a4612e9
commit 30561229c7
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
mlx/backend/cuda/distributed.cu
View File

@@ -24,6 +24,7 @@ void AllReduce::eval_gpu(
out.copy_shared_buffer(in); out.copy_shared_buffer(in);
return {in, out}; return {in, out};
} else { } else {
out.set_data(allocator::malloc(out.nbytes()));
return {in, out}; return {in, out};
} }
}; };