Skip to content

Commit 593eef1

Browse files
Update on "Add fake impl for aten.unique2"
Reapply of: #121571 Differential Revision: [D56258431](https://our.internmc.facebook.com/intern/diff/D56258431) [ghstack-poisoned]
1 parent d42e68e commit 593eef1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/test_ops.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2455,7 +2455,9 @@ def _test_fake_helper(self, device, dtype, op, context):
24552455

24562456
from torch.fx.experimental.symbolic_shapes import ShapeEnv
24572457

2458-
allow_dynamic_output_shape_shape_env = ShapeEnv(allow_dynamic_output_shape_ops=True)
2458+
allow_dynamic_output_shape_shape_env = ShapeEnv(
2459+
allow_dynamic_output_shape_ops=True
2460+
)
24592461

24602462
allow_dynamic_output_shape_mode = FakeTensorMode(
24612463
shape_env=allow_dynamic_output_shape_shape_env

0 commit comments

Comments
 (0)