double buffer size for oneprof

Dewei-Wang-sh · Sep 19, 2022 · 887839d · 887839d
1 parent d1d1c85
commit 887839d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.49.4
+0.49.5
diff --git a/tools/oneprof/finalizer.h b/tools/oneprof/finalizer.h
@@ -520,7 +520,10 @@ class Finalizer {
     PTI_ASSERT(metric_list.size() == report_size);
 
     size_t time_id = GetMetricId(metric_list, "QueryBeginTime");
-    PTI_ASSERT(time_id < metric_list.size());
+    if (time_id >= metric_list.size()) {
+      // QueryBeginTime metric does not exist
+      return 0;
+    }
 
     uint64_t max_time = (props.mask + 1ull) *
         static_cast<uint64_t>(NSEC_IN_SEC) / props.freq;

diff --git a/tools/oneprof/metric_storage.h b/tools/oneprof/metric_storage.h
@@ -15,7 +15,7 @@
 
 #define MAX_REPORT_SIZE  512
 #define MAX_REPORT_COUNT 32768
-#define MAX_BUFFER_SIZE  (MAX_REPORT_COUNT * MAX_REPORT_SIZE)
+#define MAX_BUFFER_SIZE  (MAX_REPORT_COUNT * MAX_REPORT_SIZE * 2)
 
 #define CACHE_SIZE 134217728
 
@@ -243,4 +243,4 @@ class MetricReader {
   std::vector<std::ifstream> storage_;
 };
 
-#endif // PTI_TOOLS_ONEPROF_METRIC_STORAGE_H_
+#endif // PTI_TOOLS_ONEPROF_METRIC_STORAGE_H_