@triton.jit
def store_full_1d(vals, ptr, sz: tl.constexpr, stride=1):
    """Store 1d block into vector (defined by ptr)"""
    offs = get_1d_offest(sz)
    mask = get_1d_mask(offs, sz)
    tl.store(ptr + offs, vals, mask)
