feat: make torch frame count configurable

Author: KowalskiThomas

ddtrace/internal/settings/profiling.py

@@ -495,6 +495,15 @@ class ProfilingConfigPytorch(DDConfig):
         help="How many events the PyTorch profiler records each collection",
     )
 
+    max_frames = DDConfig.v(
+        int,
+        "max_frames",
+        default=128,
+        validator=validators.range(1, t.cast(int, float("inf"))),
+        help_type="Integer",
+        help="Maximum depth to walk up the cpu_parent chain when reconstructing the operator call tree",
+    )
+
 
 class ProfilingConfigException(DDConfig):
     __item__ = __prefix__ = "exception"

ddtrace/internal/settings/profiling.pyi

@@ -52,6 +52,7 @@ class ProfilingConfigHeap(DDConfig):
 class ProfilingConfigPytorch(DDConfig):
     enabled: bool
     events_limit: int
+    max_frames: int
 
 class ProfilingConfigException(DDConfig):
     enabled: bool

ddtrace/profiling/collector/pytorch.py

@@ -18,9 +18,6 @@
 
 _NANOS_PER_MICROSECOND = 1e3
 
-# Safety bound on how far we walk up the cpu_parent chain when reconstructing
-# the operator call tree, to guard against pathological depths.
-_MAX_FRAMES = 128
 
 # Frames require a file name, but GPU frames are not from a Python file.
 # We use the following as a placeholder.
@@ -209,7 +206,7 @@ def _handle_torch_trace(prof: Any) -> None:
         handle.push_frame(e.name, _FILE_PLACEHOLDER, 0, 0)
         parent = getattr(e, "cpu_parent", None)
         depth = 0
-        while parent is not None and depth < _MAX_FRAMES:
+        while parent is not None and depth < config.pytorch.max_frames:
             handle.push_frame(parent.name, _FILE_PLACEHOLDER, 0, 0)
             parent = getattr(parent, "cpu_parent", None)
             depth += 1