Skip to main content
Log in

Mojo function

lane_group_max

lane_group_max[val_type: DType, simd_width: Int, //, nthreads: Int](val: SIMD[val_type, simd_width]) -> SIMD[val_type, simd_width]

Reduces a SIMD value to its maximum within a lane group.

Parameters:

  • val_type (DType): The type of the SIMD value.
  • simd_width (Int): The width of the SIMD value.
  • nthreads (Int): The number of threads in the warp.

Args:

  • val (SIMD[val_type, simd_width]): The SIMD value to reduce.

Returns:

A SIMD value with the maximum value in all lanes.