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][Better Engineering] Add typing annotations to guard and source (#158397)

Browse Source 
As part of better engineering week, we would like to improve out type support to improve dev experience in dynamo

This PR adds strict typing support to a critical set of files for dynamo, `source.py` and the base `_guards.py`

Running
```
mypy torch/_dynamo/source.py torch/_guards.py --linecount-report /tmp/coverage_log
```

| -------- | Lines Unannotated | Lines Total | % lines covered | Funcs Unannotated | Funcs Total | % funcs covered |
| -------- | ------- | -------- | ------- | ------- | ------- | ------- |
| Main  |  1227 | 2208 | 55.57% | 207 | 362 | 57.18% |
| This PR | 2217 | 2217 | 100.00% | 362 | 362 | 100.00% |
| Delta    | +990 | +9 | +44.43% | +155 | 0 | +42.82% |

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158397
Approved by: https://github.com/anijain2305
This commit is contained in:
Lucas Kabela
2025-07-24 15:55:18 +00:00
committed by PyTorch MergeBot
parent fd48681b6a
commit abcb24f4de
22 changed files with 335 additions and 278 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_inductor/codecache.py
View File

@ -1065,7 +1065,7 @@ class GuardedCache(Generic[T]):
Helper to get the shape env from the tracing context.
"""
ctx = torch._guards.TracingContext.try_get()
if not ctx:
if not ctx or not ctx.fake_mode:
return None
return ctx.fake_mode.shape_env