Skip to content

Commit 0cd0eec

Browse files
Lucaskabelameta-codesync[bot]
authored andcommitted
Add type hints to torch/_functorch files pt 2 (#173460)
Summary: Add comprehensive type annotations to 2 files in torch/_functorch that were previously missing type coverage. Each file now passes pyrefly check with stricter sub-config settings (implicit-any=true). Files updated: aot_autograd.py eager_transforms.py Authored with help of Claude Code X-link: pytorch/pytorch#173460 Approved by: https://github.com/aorenste Reviewed By: wdvr Differential Revision: D91862651 fbshipit-source-id: 13483cf6d59428c84e41c9d37ba8e0ed17305e5b
1 parent 29983cc commit 0cd0eec

File tree

1 file changed

+1
-1
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+1
-1
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3853,7 +3853,7 @@ def get_real_value(node: torch.fx.Node, tracer: Any) -> Any:
38533853
return real_value
38543854

38553855

3856-
def assert_no_fake_params_or_buffers(gm: torch.fx.GraphModule) -> None:
3856+
def assert_no_fake_params_or_buffers(gm: torch.nn.Module) -> None:
38573857
from torch._subclasses.fake_tensor import FakeTensorConfig, is_fake
38583858

38593859
def stack_or_hint(t: Any) -> str:

0 commit comments

Comments
 (0)