[torch/fx] add torch/utils/_stats.py to stack frame skiplist

We added some @count decorators to stuff that show up now
pytorch · Apr 1, 2023 · 0e503e8 · 0e503e8
1 parent 5df59f9
commit 0e503e8
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