Skip to content

Commit e265e37

Browse files
laithsakkafacebook-github-bot
authored andcommitted
extract shape in _view_has_unbacked_input
Summary: We were getting DDE on reshape still!! i looked deeper and found an issue in _view_has_unbacked_input namely when input is [[,,]] it need to be normalized to [..] Test Plan: existing tests. Rollback Plan: Differential Revision: D79951119
1 parent 2971231 commit e265e37

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/_subclasses/fake_impls.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -513,6 +513,7 @@ def maybe_guard_or_true(x):
513513

514514
def _view_has_unbacked_input(a, shape):
515515
from torch.fx.experimental.symbolic_shapes import has_hint
516+
shape = utils.extract_shape_from_varargs(shape, validate=False)
516517

517518
return (
518519
any(not has_hint(s) for s in a.size())

0 commit comments

Comments
 (0)