fix parsing bugs (#20246)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20246
ghimport-source-id: 611ea44dc6c1d31cee99fe3e89628be072a4a381
Differential Revision: D15247081
Pulled By: zdevito
fbshipit-source-id: 4b836ce6b665c26bb6eb5347206c55d12807fae6