zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-28 06:07:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix allocation bug in NCCL

Browse Source
This commit is contained in:
Awni Hannun 2025-08-22 14:11:52 -07:00
parent 068a4612e9
commit ef123f46e9
1 changed files with 1 additions and 0 deletions
Show Stats 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);
return {in, out};
} else {
out.set_data(allocator::malloc(out.nbytes()));
return {in, out};
}
};