--only option allows specifying which kernel

Author: gonzalobg

src/main.cpp

@@ -415,13 +415,43 @@ void parseArguments(int argc, char *argv[])
     {
       use_float = true;
     }
-    else if (!string("--triad-only").compare(argv[i]))
+    else if (!string("--print-names").compare(argv[i]))
     {
-      selection = Benchmark::Triad;
+      cout << "Available benchmarks: ";
+      print_labels(cout);
+      cout << endl;
+      exit(0);
     }
-    else if (!string("--nstream-only").compare(argv[i]))
+    else if (!string("--only").compare(argv[i]) || !string("-o").compare(argv[i]))
     {
-      selection = Benchmark::Nstream;
+      if (++i >= argc)
+      {
+        cerr << "Expected benchmark name after --only" << endl;
+        exit(1);
+      }
+      auto key = string(argv[i]);
+      if (key == "Classic")
+      {
+        selection = Benchmark::Classic;
+      }
+      else if (key == "All")
+      {
+        selection = Benchmark::All;
+      }
+      else
+      {
+        auto p = find_if(labels.begin(), labels.end(), [&](char const* label) {
+            return string(label) == key;
+          });
+        if (p == labels.end()) {
+          cerr << "Unknown benchmark name \"" << argv[i] << "\" after --only" << endl;
+          cerr << "Available benchmarks: All,Classic,";
+          print_labels(cerr);
+          cerr << endl;
+          exit(1);
+        }
+        selection = (Benchmark)(distance(labels.begin(), p));
+      }
     }
     else if (!string("--csv").compare(argv[i]))
     {
@@ -443,8 +473,8 @@ void parseArguments(int argc, char *argv[])
       cout << "  -s  --arraysize  SIZE    Use SIZE elements in the array" << endl;
       cout << "  -n  --numtimes   NUM     Run the test NUM times (NUM >= 2)" << endl;
       cout << "      --float              Use floats (rather than doubles)" << endl;
-      cout << "      --triad-only         Only run triad" << endl;
-      cout << "      --nstream-only       Only run nstream" << endl;
+      cout << "  -o  --only       NAME    Only run one benchmark (see --print-names)" << endl;
+      cout << "      --print-names        Prints all available benchmark names" << endl;
       cout << "      --csv                Output as csv table" << endl;
       cout << "      --mibibytes          Use MiB=2^20 for bandwidth calculation (default MB=10^6)" << endl;
       cout << endl;