Make the script more robust on supporting unknown devices (#642)
Summary:
Only do device-check on CPUs.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/642
Reviewed By: xuzhao9
Differential Revision: D33072444
Pulled By: alanwaketan
fbshipit-source-id: e5cc16bc74658b37341982a8810ccdde26f2cb6e