Mojo function
max
max[type: DType, width: Int, //, *, block_size: Int](val: SIMD[type, width]) -> SIMD[type, width]
Computes the maximum value across all threads in a block.
Performs a parallel reduction using warp-level operations and shared memory to find the global maximum across all threads in the block.
Parameters:
- type (
DType): The data type of the SIMD elements. - width (
Int): The number of elements in each SIMD vector. - block_size (
Int): The total number of threads in the block.
Args:
- val (
SIMD[type, width]): The SIMD value to reduce. Each thread contributes its value to find the maximum.
Returns:
A SIMD value where all threads contain the maximum value found across the entire block.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!