Limit TSL stack size to 10

PiperOrigin-RevId: 691971293
indhub · Oct 31, 2024 · d5a760b · d5a760b
1 parent e638627
commit d5a760b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/xla/service/gpu/metrics.cc b/xla/service/gpu/metrics.cc
@@ -112,6 +112,11 @@ void RecordGpuCompilerStacktrace() {
   stack.pop_front();
   stack.pop_back();
 
+  const int kMaxStackDepth = 10;
+  while (stack.size() > kMaxStackDepth) {
+    stack.pop_back();
+  }
+
   // Stack traces with addresses would make too many unique streamz cells.
   // We only care about the actual call stack.
   // Format chars added by tsl::CurrentStackTrace().