Mojo function
rv_all_below
rv_all_below[layout_rv: TensorLayout, //](max_prev: TileTensor[DType.float32, layout_rv, MutExternalOrigin, address_space=AddressSpace.LOCAL], max_new: TileTensor[DType.float32, layout_rv, MutExternalOrigin, address_space=AddressSpace.LOCAL], threshold: Float32) -> Bool
Returns wave-uniform True iff every lane satisfies max_new - max_prev <= threshold. Wave AND-reduce via a 64-bit ballot compared against the full-exec mask (attend_ker always runs all 64 lanes active). Used by the lazy-rescale skip path.
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!