Skip to main content

Module capabilities

Module capabilities 

Source
Expand description

Runtime probe for NCCL capabilities.

Surfaces version + opt-in feature gates: fp8 reduction (NCCL >= 2.20), NVLS (NCCL >= 2.18 on supported topologies), SHARP. The probe is best-effort: if NCCL isn’t loadable on this host (e.g. a CPU-only CI runner), the probe returns NcclCapabilities::zeroed rather than panicking.

Structs§

NcclCapabilities
Static description of the loaded NCCL library’s capabilities.

Functions§

probe_capabilities
Best-effort capability probe. Calls ncclGetVersion via cudarc’s safe wrapper; on any error returns NcclCapabilities::zeroed.