Add --grid-only switch to the benchmark script