AliceO2Group
diff --git a/‎GPU/GPUbenchmark/Shared/Utils.h‎
Lines changed: 1 addition & 0 deletions b/‎GPU/GPUbenchmark/Shared/Utils.h‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎GPU/GPUbenchmark/benchmark.cxx‎
Lines changed: 5 additions & 2 deletions b/‎GPU/GPUbenchmark/benchmark.cxx‎
Lines changed: 5 additions & 2 deletions
@@ -75,6 +75,7 @@ struct benchmarkOpts {
   float freeMemoryFractionToAllocate = 0.95f;
   int kernelLaunches = 1;
   int nTests = 1;
+  std::string outFileName = "benchmark_result";
 };
 
 template <class T>
 
@@ -28,7 +28,8 @@ bool parseArgs(o2::benchmark::benchmarkOpts& conf, int argc, const char* argv[])
     "chunkSize,c", bpo::value<float>()->default_value(1.f), "Size of scratch partitions (GB).")(
     "freeMemFraction,f", bpo::value<float>()->default_value(0.95f), "Fraction of free memory to be allocated (min: 0.f, max: 1.f).")(
     "launches,l", bpo::value<int>()->default_value(10), "Number of iterations in reading kernels.")(
-    "nruns,n", bpo::value<int>()->default_value(1), "Number of times each test is run.");
+    "nruns,n", bpo::value<int>()->default_value(1), "Number of times each test is run.")(
+    "outfile,o", bpo::value<std::string>()->default_value("benchmark_result"), "Output file name to store results.");
   try {
     bpo::store(parse_command_line(argc, argv, options), vm);
     if (vm.count("help")) {
@@ -89,6 +90,8 @@ bool parseArgs(o2::benchmark::benchmarkOpts& conf, int argc, const char* argv[])
     }
   }
 
+  conf.outFileName = vm["outfile"].as<std::string>();
+
   return true;
 }
 
@@ -103,7 +106,7 @@ int main(int argc, const char* argv[])
     return -1;
   }
 
-  std::shared_ptr<ResultWriter> writer = std::make_shared<ResultWriter>(std::to_string(opts.deviceId) + "_benchmark_results.root");
+  std::shared_ptr<ResultWriter> writer = std::make_shared<ResultWriter>(std::to_string(opts.deviceId) + "_" + opts.outFileName + ".root");
 
   o2::benchmark::GPUbenchmark<char> bm_char{opts, writer};
   bm_char.run();
Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,8 @@ bool parseArgs(o2::benchmark::benchmarkOpts& conf, int argc, const char* argv[])`
`28`	`28`	`"chunkSize,c", bpo::value<float>()->default_value(1.f), "Size of scratch partitions (GB).")(`
`29`	`29`	`"freeMemFraction,f", bpo::value<float>()->default_value(0.95f), "Fraction of free memory to be allocated (min: 0.f, max: 1.f).")(`
`30`	`30`	`"launches,l", bpo::value<int>()->default_value(10), "Number of iterations in reading kernels.")(`
`31`		`- "nruns,n", bpo::value<int>()->default_value(1), "Number of times each test is run.");`
	`31`	`+ "nruns,n", bpo::value<int>()->default_value(1), "Number of times each test is run.")(`
	`32`	`+ "outfile,o", bpo::value<std::string>()->default_value("benchmark_result"), "Output file name to store results.");`
`32`	`33`	`try {`
`33`	`34`	`bpo::store(parse_command_line(argc, argv, options), vm);`
`34`	`35`	`if (vm.count("help")) {`
`@@ -89,6 +90,8 @@ bool parseArgs(o2::benchmark::benchmarkOpts& conf, int argc, const char* argv[])`
`89`	`90`	`}`
`90`	`91`	`}`
`91`	`92`
	`93`	`+ conf.outFileName = vm["outfile"].as<std::string>();`
	`94`	`+`
`92`	`95`	`return true;`
`93`	`96`	`}`
`94`	`97`
`@@ -103,7 +106,7 @@ int main(int argc, const char* argv[])`
`103`	`106`	`return -1;`
`104`	`107`	`}`
`105`	`108`
`106`		`- std::shared_ptr<ResultWriter> writer = std::make_shared<ResultWriter>(std::to_string(opts.deviceId) + "_benchmark_results.root");`
	`109`	`+ std::shared_ptr<ResultWriter> writer = std::make_shared<ResultWriter>(std::to_string(opts.deviceId) + "_" + opts.outFileName + ".root");`
`107`	`110`
`108`	`111`	`o2::benchmark::GPUbenchmark<char> bm_char{opts, writer};`
`109`	`112`	`bm_char.run();`