zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-24 22:36:39 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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); 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};
} }
}; };