8000 . · pytorch/pytorch@9b35b2e · GitHub
[go: up one dir, main page]
Skip to content

Commit 9b35b2e

Browse files
pianpwkpianpwk
committed
.
1 parent 40303c5 commit 9b35b2e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

‎torch/fx/experimental/symbolic_shapes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4501,7 +4501,8 @@ def create_symbol(
45014501
SymT.SIZE, len(self.var_to_val), positive=positive, integer=True
45024502
)
45034503
if (
4504-
isinstance(source, TensorPropertySource)
4504+
not specialize_zero_one
4505+
and isinstance(source, TensorPropertySource)
45054506
and source.prop == TensorProperty.SIZE
45064507
):
45074508
self.size_like.add(sympy_expr)

0 commit comments

Comments
 (0)
0