Change localrun to cpu_only and mention specifying this option will a…

…utomatically ignore the machine config check.

README.md

@@ -91,7 +91,7 @@ Some useful options include
 - `--benchmark-autosave` (or other save related flags) to get .json output
 - `-k <filter expression>` (standard pytest filtering)
 - `--collect-only` only show what tests would run, useful to see what models there are or debug your filter expression
-- `--localrun` if running on a CPU machine
+- `--cpu_only` if running on a local CPU machine and ignoring machine configuration checks
 
 ## Nightly CI runs
 Currently, models run on nightly pytorch builds and push data to scuba.

conftest.py

@@ -14,8 +14,8 @@ def pytest_addoption(parser):
  parser.addoption("--check_opt_vs_noopt_jit",
  action='store_true',
  help="The best attempt to check results for inference runs. Not all models support this!")
- parser.addoption("--localrun", action='store_true',
- help="Run benchmarks locally on a macbook")
+ parser.addoption("--cpu_only", action='store_true',
+ help="Run benchmarks on cpu only and ignore machine configuration checks")
 
 def set_fuser(fuser):
  if fuser == "old":

test_bench.py

@@ -24,7 +24,7 @@ def pytest_generate_tests(metafunc):
  # This is where the list of models to test can be configured
  # e.g. by using info in metafunc.config
  devices = ['cpu', 'cuda']
- if metafunc.config.option.localrun:
+ if metafunc.config.option.cpu_only:
  devices = ['cpu']
  all_models = list_models()
  if metafunc.cls and metafunc.cls.__name__ == "TestBenchNetwork":