We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4ed2ef6 commit 2dbbe01Copy full SHA for 2dbbe01
test/test_dynamic_shapes.py
@@ -11,6 +11,7 @@
11
12
class TestDynamicShapes(unittest.TestCase):
13
14
+ @unittest.skip("Needs to lower ne")
15
def test_simple_expand(self):
16
size1 = 5
17
size2 = 2
@@ -25,6 +26,7 @@ def test_simple_expand(self):
25
26
t6_cpu = t6.cpu()
27
self.assertEqual(t6_cpu.shape[0], 2)
28
29
30
def test_simple_expand_on_2d_tensor(self):
31
32
@@ -66,6 +68,7 @@ def test_wrap(self):
66
68
a3 = a2.shape[0] + 3 # tests wrap
67
69
self.assertIsInstance(a3, torch.SymInt)
70
71
72
def test_sizeAdd(self):
73
74
0 commit comments