Skip to main content

Module nccl_stubs

Module nccl_stubs 

Source
Expand description

Stub types for NCCL when the cuda feature is disabled.

These types match the public API of the real nccl module so that downstream crates (e.g., monarch_tensor_worker) compile without CUDA. Constructors and operations panic at runtime; the types exist solely for type-level compatibility, and callers are expected to gate usage on a tensor-engine capability check before reaching this code.

Re-exports§

pub use crate::nccl_common::DataType;
pub use crate::nccl_common::NcclError;
pub use crate::nccl_common::NcclGroupTicket;
pub use crate::nccl_common::NcclStatus;
pub use crate::nccl_common::RawNcclError;

Structs§

Communicator
UniqueId
Binding for ncclUniqueId.

Enums§

ReduceOp
Rust version of ncclRedOp_t.

Traits§

UniqueIdExt
Extension trait providing NCCL-specific operations on UniqueId.

Functions§

group_end
group_start