Skip to content

Commit

Permalink
added comma for reformatting
Browse files Browse the repository at this point in the history
  • Loading branch information
@brucekimrokcmu
brucekimrokcmu committed Jul 25, 2023
1 parent a4d4e34 commit 58f80d8
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions tests/unit/test_list_int_bindings.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,35 +18,35 @@ def test_max_pool2d(self):
l = ops.max_pool2d(t, 3, 1)
check_correct(
" %4 = torch.aten.max_pool2d %arg0, %1, %2, %3, %0, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l = ops.max_pool2d(t, (3, 3), 1)
check_correct(
" %4 = torch.aten.max_pool2d %arg0, %1, %2, %3, %0, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l = ops.max_pool2d(t, 3, (2, 2))
check_correct(
" %4 = torch.aten.max_pool2d %arg0, %1, %2, %3, %0, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l = ops.max_pool2d(t, 3, 1, 2, 3)
check_correct(
" %4 = torch.aten.max_pool2d %arg0, %1, %2, %3, %0, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l = ops.max_pool2d(t, 3, 1, [2, 2], 3)
check_correct(
" %4 = torch.aten.max_pool2d %arg0, %1, %2, %3, %0, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor",
l.owner
l.owner,
)

def test_max_pool2d_with_indices(self):
Expand All @@ -55,26 +55,26 @@ def test_max_pool2d_with_indices(self):
l, s = ops.max_pool2d_with_indices(t, 3)
check_correct(
"%result0, %result1 = torch.aten.max_pool2d_with_indices %arg0, %0, %1, %2, %1, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor, !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l, s = ops.max_pool2d_with_indices(t, 3, (1, 1), 2)
check_correct(
"%result0, %result1 = torch.aten.max_pool2d_with_indices %arg0, %0, %1, %2, %1, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor, !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l, s = ops.max_pool2d_with_indices(t, 3, (1, 1), 2, [4, 4])
check_correct(
"%result0, %result1 = torch.aten.max_pool2d_with_indices %arg0, %0, %1, %2, %1, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor, !torch.tensor",
l.owner
l.owner,
)

t = pi.rand(1, 3, 32, 32)
l, s = ops.max_pool2d_with_indices(t, 3, 1, 2, 4)
check_correct(
"%result0, %result1 = torch.aten.max_pool2d_with_indices %arg0, %0, %1, %2, %1, %false : !torch.tensor<[1,3,32,32],f64>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.list<int>, !torch.bool -> !torch.tensor, !torch.tensor",
l.owner
l.owner,
)

0 comments on commit 58f80d8

Please sign in to comment.