@triton.jit
def find_kernel(parent, u, ret_ptr, BLOCK_SIZE: tl.constexpr):
    pu = tl.load(parent + u)
    while pu != u:
        u = pu
        pu = tl.load(parent + u)
    ret_ptr[u % BLOCK_SIZE] = pu
