Module nccl
Source - Communicator
- Wraps a NCCL communicator, and provides a Tensor-based interface it.
- NcclGroupTicket
- A ticket that we use to link group start/end calls. Does not implement
Send, to enforce that group start and end calls are on the same thread. - UniqueId
- Binding for
ncclUniqueId.
- DataType
- Rust version of
ncclDataType_t. - NcclError
- Types of errors that the safe
Communicator API can return. - NcclStatus
- RawNcclError
- Corresponds to ncclResult_t error cases
- ReduceOp
- Rust version of
ncclRedOp_t.
- group_end
- End the NCCL group.
- group_start
- Start a new NCCL group. All NCCL calls within this group will be combined,
provided that they were issued on the same thread.