Add the --mps_only option. (#1432)
Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1432
Reviewed By: davidberard98
Differential Revision: D43587178
Pulled By: xuzhao9
fbshipit-source-id: d10ed14aca5a17ceea29ac680351120477fbcd41
