Trait NcclReduceSupported 

pub trait NcclReduceSupported:
    NcclType
    + Copy
    + Send
    + Sync
    + 'static {
    // Required method
    fn dispatch_dtype() -> DispatchDType;
}

Marker for dtypes carried in NCCL collectives. Mirrors the NcclReduceSupported marker that the Phase 0 dtype.rs will host — defined locally here so the NCCL slice can ship before Phase 0 fully lands. The set matches NCCL’s reduce-supported types: f32, f64, f16, bf16, i8, u8, i32, u32, i64, u64. fp8 e4m3/e5m2 are behind nccl-fp8 and require NCCL >= 2.20.

Required Methods

fn dispatch_dtype() -> DispatchDType

Static dtype tag for tracing.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types

impl NcclReduceSupported for f32

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for f64

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for i8

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for i32

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for i64

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for u8

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for u32

fn dispatch_dtype() -> DispatchDType

impl NcclReduceSupported for u64

fn dispatch_dtype() -> DispatchDType

Implementors