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

[dynamo][refactor][config-cleanp] Use guard_manager consistently instead of check_fn (#138896)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138896
Approved by: https://github.com/williamwen42, https://github.com/jansel
ghstack dependencies: #138512
This commit is contained in:
Animesh Jain
2024-10-25 12:14:35 -07:00
committed by PyTorch MergeBot
parent 49ed365b22
commit dba6887dc6
10 changed files with 78 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_dynamo/testing.py
View File

@ -191,7 +191,7 @@ def debug_insert_nops(
torch_function_mode_stack=[],
)
return GuardedCode(code, CheckFunctionManager(graph).check_fn, CompileId(0, 0))
return GuardedCode(code, CheckFunctionManager(graph).guard_manager, CompileId(0, 0)) # type: ignore[arg-type]
class CompileCounter: