@triton.jit
def ub2(X, Y):
    r = tl.arange(0, 16)
    r2 = tl.arange(0, 32)
    x = tl.load(X + 16 * r2[:, None] + r)
    y = triton_unbroadcast(x, tl.arange(0, 32)[:, None].shape)
    tl.store(Y + r2[:, None], y)
