[torch/fx] add torch/utils/_stats.py to stack frame skiplist (#98117)

We added some @count decorators to stuff that show up now Pull Request resolved: #98117 Approved by: https://github.com/SherlockNoMad
pytorch · Apr 4, 2023 · 8f2f1a0 · 8f2f1a0
1 parent 1fae179
commit 8f2f1a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch/fx/proxy.py b/torch/fx/proxy.py
@@ -201,7 +201,8 @@ def _find_user_frame(self):
                     'torch/utils/_python_dispatch.py',
                     'torch/_prims_common/wrappers.py',
                     'torch/_refs/__init__.py',
-                    'torch/_refs/nn/functional/__init__.py'
+                    'torch/_refs/nn/functional/__init__.py',
+                    'torch/utils/_stats.py',
                     ]
         while frame:
             frame = frame.f_back