Module nccl

Source

Structs§

Communicator
Wraps a NCCL communicator, and provides a Tensor-based interface it.
NcclConfig
Rust version of ncclConfig_t. See nccl documentation for what each field means: https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/api/types.html#ncclconfig-t
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.

Enums§

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.

Functions§

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.