Porting fixes from wconstab/ltc to main (#851)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/851
Reviewed By: pbelevich
Differential Revision: D35420935
Pulled By: Krovatkin
fbshipit-source-id: 60eb25c33956e15d12fdac54b5eda21e25ba2aa7