Module nccl Copy item path Source 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
. 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.