Skip to content

Commit 2dead01

Browse files
committed
add test cases for empty dims
1 parent 8d968cf commit 2dead01

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/py/dynamo/conversion/test_flip_aten.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ class TestFlipConverter(DispatchTestCase):
1111
[
1212
((3,), [0]),
1313
((3,), [-1]),
14+
((3,), []),
1415
((3, 3), [0, 1]),
1516
((3, 3), [-2, 1]),
1617
((2, 3, 4), [0]),
@@ -20,6 +21,7 @@ class TestFlipConverter(DispatchTestCase):
2021
((3, 3, 3, 3), [0]),
2122
((2, 3, 4, 5), [0, 1, 2, 3]),
2223
((2, 3, 4, 5), [-4, 1, -2, 3]),
24+
((2, 3, 4, 5), []),
2325
]
2426
)
2527
def test_flip(self, shape, dims):

0 commit comments

Comments
 (0)