Some minor fixes for the changes in #20945 (#21008)
Summary:
Fixes after #20945
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21008
Differential Revision: D15526193
Pulled By: ezyang
fbshipit-source-id: 4cfabc482c149e0aeb92ae7fff04098771fe33ed
