diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py index 7f06d7eab8..7d8a2becb0 100644 --- a/thunder/tests/opinfos.py +++ b/thunder/tests/opinfos.py @@ -436,9 +436,8 @@ def elementwise_unary_generator( ) shapes = ( - # TODO: restore size zero cases - # (0, 2, 1), - # (5, 0, 3), + (0, 2, 1), + (5, 0, 3), (), (11,), (4, 4),