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 7c69c31 commit 8831462Copy full SHA for 8831462
tests/py/dynamo/conversion/test_local_scalar_dense_aten.py
@@ -8,24 +8,12 @@
8
class TestLocalScalarDenseConverter(DispatchTestCase):
9
@parameterized.expand(
10
[
11
- (
12
- torch.randn((5, 10, 5), dtype=torch.float32),
13
- ),
14
15
- torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32),
16
17
18
- torch.randn((1), dtype=torch.float32),
19
20
21
- (torch.tensor([-2.4])),
22
23
24
- (torch.tensor([5.5, 3.5, 3.6])),
25
26
27
- (torch.tensor([True])),
28
+ (torch.randn((5, 10, 5), dtype=torch.float32),),
+ (torch.randint(-10, 10, (5, 1, 15), dtype=torch.int32),),
+ (torch.randn((1), dtype=torch.float32),),
+ ((torch.tensor([-2.4])),),
+ ((torch.tensor([5.5, 3.5, 3.6])),),
+ ((torch.tensor([True])),),
29
(
30
torch.tensor(
31
@@ -44,9 +32,7 @@ class TestLocalScalarDenseConverter(DispatchTestCase):
44
32
]
45
33
),
46
34
47
48
- (torch.tensor([float("inf")])),
49
35
+ ((torch.tensor([float("inf")])),),
50
36
51
37
)
52
38
def test_local_scalar_dense(self, data):
0 commit comments