Checkout the lazy_tensor_staging branch from the pytorch repo. (#615)
Summary:
Fixes lazy tensor nightly CI.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/615
Reviewed By: alanwaketan
Differential Revision: D32895074
Pulled By: xuzhao9
fbshipit-source-id: de8287fd6db8d6951c40d9714768ba4d9478bead