frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dynamo] Add recompile reason for set_stance fail_on_recompile (#165445)

Browse Source 
Fixes #163500

### Summary:
For `set_stance("fail_on_recompile")` failures will provide the reason why the recompilation occurred

### Impacts:
module: dynamo

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165445
Approved by: https://github.com/williamwen42
This commit is contained in:
Parshant Sharma
2025-10-19 21:12:16 +00:00
committed by PyTorch MergeBot
parent a88587348b
commit 8139f33fa5
2 changed files with 41 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
torch/_dynamo/eval_frame.py
View File

@ -235,7 +235,11 @@ def _callback_from_stance(callback: DynamoCallback) -> DynamoCallback:
if not convert_frame.has_tensor_in_frame(frame):
return ConvertFrameReturn()
from torch._C._dynamo.eval_frame import _debug_get_precompile_entries
from torch._C._dynamo.eval_frame import (
_debug_get_cache_entry_list,
_debug_get_precompile_entries,
)
from torch._dynamo.guards import get_and_maybe_log_recompilation_reasons
message = (
"Detected recompile when torch.compile stance is 'fail_on_recompile'. "
@ -243,6 +247,17 @@ def _callback_from_stance(callback: DynamoCallback) -> DynamoCallback:
+ f"function name: '{frame.f_code.co_name}', "
+ f"line number: {frame.f_lineno}"
)
cache_entries = _debug_get_cache_entry_list(frame.f_code)
if cache_entries:
reasons = get_and_maybe_log_recompilation_reasons(
cache_entries[0], frame, skip_logging=True
)
if reasons:
failures = textwrap.indent("\n".join(reasons), "- ")
guard_failure_details = (
f"triggered by the following guard failure(s):\n{failures}"
)
message += f"\n{textwrap.indent(guard_failure_details, ' ')}"
precompile_entries = _debug_get_precompile_entries(frame.f_code)
if len(precompile_entries) > 0:
message += "\nFailed on the following precompiled guards: "