Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

what is the count value in allreduce function? #1508

Open
BSkim26 opened this issue Nov 8, 2024 · 0 comments
Open

what is the count value in allreduce function? #1508

BSkim26 opened this issue Nov 8, 2024 · 0 comments

Comments

@BSkim26
Copy link

BSkim26 commented Nov 8, 2024

I'm studying about the NCCL, and I am wondering what is the the count value in allreduce function.

ncclResult_tncclAllReduce(const void* sendbuff, void* recvbuff, size_t count, ncclDataType_t datatype, ncclRedOp_t op, ncclComm_t comm, cudaStream_t stream)

if I understand correctly, sendbuff and recvbuff are size of the transfered data for allreduce, and count value is the number of float data units.
in the ring allreudce, I think, we have to transfer data n-1 times per each GPU.
so the whole data leaving each gpu is (n-1)countfloat = (n-1)*sendbuff=(n-1)*recvbuff?

and then in the formula (algbw = S/t), the Size, S, is same as the count or # of GPUs * count * float?

https://github.com/NVIDIA/nccl-tests/blob/master/doc/PERFORMANCE.md#bandwidth

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant