Minor preparational JIT changes (#21096)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21096
ghimport-source-id: 169f8b4b70cd77b0f9b07cca81d2b4cde2c46456
Reviewed By: ezyang
Differential Revision: D15546176
Pulled By: izdeby
fbshipit-source-id: cdd0a1c87263955eef9d3174ec2f36d1d2935f48
