Mojo struct
ReduceScatterConfig
@register_passable(trivial)
struct ReduceScatterConfig[dtype: DType, ngpus: Int, simd_width: Int = simd_width_of[dtype, get_gpu_target()](), alignment: Int = align_of[SIMD[dtype, simd_width]](), accum_type: DType = get_accum_type[dtype]()]
Fields
- stride (
Int): - largest_part (
Int): - part (
Int): - rank_start (
Int): - rank_end (
Int): - thr_local_start (
Int):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable
comptime members
__copyinit__is_trivial
comptime __copyinit__is_trivial = True
__del__is_trivial
comptime __del__is_trivial = True
__moveinit__is_trivial
comptime __moveinit__is_trivial = True
Methods
__init__
__init__(num_elements: Int, thread_idx: Int, threads_per_gpu: Int, my_rank: Int) -> Self
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!