Fix the return value of select_device() (#1586)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1586
Reviewed By: janeyx99
Differential Revision: D45341227
Pulled By: xuzhao9
fbshipit-source-id: 26776d8f1fdf5f137a465672219f9e4293735c04