taoshan: fixup OmniTorch