OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[export] Allow user frame to be None when symbolic shape tries to get stacktrace. (#155744)

Browse Source 
Summary: Fixing https://github.com/pytorch/pytorch/issues/155605

Test Plan:
CI

Rollback Plan:

Differential Revision: D76463358

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155744
Approved by: https://github.com/angelayi
This commit is contained in:
Zhengxu Chen 2025-06-12 14:36:29 +00:00 committed by PyTorch MergeBot
parent dd1b6621bc
commit 0fd711df19
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
torch/fx/experimental/symbolic_shapes.py
View File

@ -7003,13 +7003,12 @@ class ShapeEnv:
stack_info=True if log.getEffectiveLevel() < logging.WARNING else False,
)
def _get_user_frame(self) -> types.FrameType:
def _get_user_frame(self) -> Optional[types.FrameType]:
frame = inspect.currentframe()
while frame is not None:
if frame.f_code.co_filename not in uninteresting_files():
return frame
frame = frame.f_back
assert frame is not None
return frame
def _get_stack_summary(
@ -7019,11 +7018,12 @@ class ShapeEnv:
if floc is None:
frame = self._get_user_frame()
try:
floc = traceback.FrameSummary(
frame.f_code.co_filename,
frame.f_lineno,
frame.f_code.co_name,
)
if frame is not None:
floc = traceback.FrameSummary(
frame.f_code.co_filename,
frame.f_lineno,
frame.f_code.co_name,
)
finally:
del frame