For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo struct
ReduceMin
struct ReduceMin[dtype: DType, W: Int = simd_width_of[dtype]()]
Min reduction monoid: (self, x) -> min(self, x).
Single-field state: acc: SIMD[dtype, W]. accumulate is one
expression β acc = min(acc, Self.pad[...](val)). The horizontal
collapse to scalar happens once at reduce, in acc[0].
Parametersβ
- βdtype (
DType): The accumulator dtype. - βW (
Int): SIMD width of the lane-wise accumulator. Defaults to the target'ssimd_width_of[dtype].
Fieldsβ
- βacc (
SIMD[dtype, W]): Lane-wise SIMD accumulator.reducereduces to scalar (placed inacc[0]); bodies readacc[0].
Implemented traitsβ
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
ReduceOp,
RegisterPassable,
TrivialRegisterPassable
comptime membersβ
Singleβ
comptime Single = ReduceMin[dtype, Int(1)]
widthβ
comptime width = W
Methodsβ
__init__β
def __init__() -> Self
Identity: acc = MAX_W (True for bool: min is logical AND).
__getitem__β
def __getitem__(self, j: Int) -> Self.Single
Returns lane j as a width-1 monoid.
Returns:
Self.Single
__setitem__β
def __setitem__(mut self, j: Int, s: ReduceMin[dtype, Int(1)])
Writes width-1 monoid s into lane j.
accumulateβ
def accumulate[val_dtype: DType, w: Int](mut self, val: SIMD[val_dtype, w], idx: SIMD[DType.int64, w] = 0)
Folds a SIMD tile into acc lane-wise. Partial tiles get identity-padded (max_finite) via Self.pad.
Parameters:
Args:
- βval (
SIMD[val_dtype, w]): The SIMD tile to fold. - βidx (
SIMD[DType.int64, w]): Unused (Min is index-agnostic).
joinβ
def join(mut self, other: Self)
Sequential combine: lane-wise min.
Args:
- βother (
Self): The state to combine intoself.
reduceβ
def reduce(self) -> Self.Single
Mins the W lane partials via the reduce_min intrinsic.
Overrides the default lane-fold so the vectorized horizontal min
is emitted explicitly (a measured CPU win; GPU-neutral). bool
reduces through uint8 because the FP reduce_min intrinsic
rejects <N x i1> on GPU.
Returns:
Self.Single: A width-1 ReduceMin holding the minimum in acc[0].
join_parallelβ
def join_parallel[R: Reducer](mut self, reducer: R)
Cross-thread combine via reducer.min, splatting the scalar result across all lanes. Runs after reduce.
Parameters:
- βR (
Reducer): The parallel scalar reducer.
Args:
- βreducer (
R): The reducer instance.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!