Mojo function
lane_group_sum
lane_group_sum[val_type: DType, simd_width: Int, //, nthreads: Int](val: SIMD[val_type, simd_width]) -> SIMD[val_type, simd_width]
Computes the sum 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 sum.
Returns:
A SIMD value with the sum of all lanes.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!