Skip to content

Commit

Permalink
Make _inductor/fx_utils.py, _dynamo/utils.py pass follow_imports type…
Browse files Browse the repository at this point in the history
…checking (#113722)

Summary:
X-link: pytorch/pytorch#113722
Approved by: https://github.com/lezcano

Reviewed By: huydhn

Differential Revision: D51418061

Pulled By: int3

fbshipit-source-id: 1784dab4e5a908b1c45543b8e8785abea87b217e
  • Loading branch information
int3 authored and facebook-github-bot committed Nov 17, 2023
1 parent 4278545 commit 4f134a4
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1933,6 +1933,8 @@ def __call__(self, *args, **kwargs):
def defake(x):
if not isinstance(x, FakeTensor):
return x
size: "torch._prims_common.ShapeType"
stride: "torch._prims_common.StrideType"
if x._has_symbolic_sizes_strides:
size = [
s.node.shape_env.size_hint(s.node.expr)
Expand Down

0 comments on commit 4f134a4

Please sign in to comment.