Adjust GetMeanTime method for runs in validate mode

7a5be097 · Ramesh Errabolu · 1d123118 · 7a5be097
Commit 7a5be097 authored Jun 19, 2019 by Ramesh Errabolu
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

rocm_bandwidth_test_report.cpp rocm_bandwidth_test_report.cpp +5 -0

No files found.
--- a/rocm_bandwidth_test_report.cpp
+++ b/rocm_bandwidth_test_report.cpp
@@ -124,6 +124,11 @@ double RocmBandwidthTest::GetMinTime(std::vector<double>& vec) {
 double RocmBandwidthTest::GetMeanTime(std::vector<double>& vec) {
+  // In validation mode we run only one iteration
+  if (validate_) {
+    return vec.at(0);
+  }
  std::sort(vec.begin(), vec.end());
  vec.erase(vec.begin());
  vec.erase(vec.begin(), vec.begin() + num_iteration_ * 0.1);