@triton.jit
def add_kernel(x, y, out, n, BLOCK_SIZE: tl.constexpr):
    start = BLOCK_SIZE * tl.program_id(axis=0)
    offsets = start + tl.arange(0, BLOCK_SIZE)
    mask = offsets < n
    xs = tl.load(x + offsets, mask=mask)
    ys = tl.load(y + offsets, mask=mask)
    tl.store(out + offsets, xs + ys, mask=mask)
