[inductor] Make graph.py pass follow_imports typechecking (#113518)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/113518
Approved by: https://github.com/Skylion007
ghstack dependencies: #113413
diff --git a/torch/_C/__init__.pyi.in b/torch/_C/__init__.pyi.in
index f351ad5..48a4651 100644
--- a/torch/_C/__init__.pyi.in
+++ b/torch/_C/__init__.pyi.in
@@ -1593,6 +1593,7 @@
_backward_hooks: Optional[Dict[_int, Callable[[Tensor], Optional[Tensor]]]]
nbytes: _int
itemsize: _int
+ _has_symbolic_sizes_strides: _bool
${tensor_method_hints}
_TensorBase = TensorBase
diff --git a/torch/_inductor/utils.py b/torch/_inductor/utils.py
index ce42cbd..9772fe4 100644
--- a/torch/_inductor/utils.py
+++ b/torch/_inductor/utils.py
@@ -233,7 +233,9 @@
return n
-def convert_shape_to_inductor(lst: List[Union[int, torch.SymInt]]) -> List[sympy.Expr]:
+def convert_shape_to_inductor(
+ lst: Iterable[Union[int, torch.SymInt]]
+) -> List[sympy.Expr]:
"""
Gets the shape and stride of a tensor. For non-symbolic tensors, this is
trivial. But for symbolic tensors, we need to map from SymIntNode into
