modify: change the buffer shape (size_buf*nlev, num_var) -> (nlev, size_buf, num_var)