Skip to content

Commit

Permalink
Revert "disable conv3d test (#7969)"
Browse files Browse the repository at this point in the history
This reverts commit cbaa81d.
  • Loading branch information
jackalcooper committed Apr 7, 2022
1 parent cbaa81d commit 34cc223
Showing 1 changed file with 7 additions and 8 deletions.
15 changes: 7 additions & 8 deletions python/oneflow/test/modules/test_conv3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,14 +22,13 @@

@flow.unittest.skip_unless_1n1d()
class TestConv3DModule(flow.unittest.TestCase):
# Disable this test for unknown error
# @autotest(n=3)
# def test_nn_functional_conv3d(test_case):
# device = random_device()
# img = torch.ones((1, 3, 224, 224, 224), requires_grad=True).to(device)
# kernel = torch.ones((6, 3, 3, 3, 3), requires_grad=True).to(device)
# y = torch.nn.functional.conv3d(img, kernel)
# return y
@autotest(n=3)
def test_nn_functional_conv3d(test_case):
device = random_device()
img = torch.ones((1, 3, 224, 224, 224), requires_grad=True).to(device)
kernel = torch.ones((6, 3, 3, 3, 3), requires_grad=True).to(device)
y = torch.nn.functional.conv3d(img, kernel)
return y

@autotest(n=10)
def test_conv3d_with_random_data(test_case):
Expand Down

0 comments on commit 34cc223

Please sign in to comment.