Add new autotest

oneflow/python/nn/modules/squeeze.py

@@ -63,7 +63,6 @@ def squeeze_op(input, dim: Optional[Sequence[int]] = None):
         dim = [dim]
     elif dim is None:
         dim = range(input.ndim)
-
     dim = list(filter(lambda i: input.size(i) == 1, dim))
     return Squeeze(dim=dim)(input)
 

oneflow/python/nn/modules/transpose.py

@@ -22,7 +22,11 @@
 
 class Transpose(Module):
     def __init__(
-        self, dim0, dim1, conjugate: bool = False, batch_axis_non_change: bool = False,
+        self,
+        dim0: int,
+        dim1: int,
+        conjugate: bool = False,
+        batch_axis_non_change: bool = False,
     ) -> None:
         super().__init__()
 

oneflow/python/test/modules/test_activation.py

@@ -737,9 +737,15 @@ def test_hardswish(test_case):
         for arg in GenArgList(arg_dict):
             _test_hardswish_impl(test_case, *arg)
 
+    @autotest()
     def test_hardswish_module_with_random_data(test_case):
-        for device in ["cpu", "cuda"]:
-            test_module_against_pytorch(test_case, "nn.Hardswish", device=device, n=2)
+        m = torch.nn.Hardswish()
+        m.train(random())
+        device = random_device()
+        m.to(device)
+        x = random_pytorch_tensor().to(device)
+        y = m(x)
+        return y
 
 
 def _np_hardtanh_grad(x):

oneflow/python/test/modules/test_prelu.py

@@ -20,6 +20,7 @@
 
 import oneflow as flow
 from test_util import GenArgList
+from automated_test_util import *
 
 
 def _prelu(input, alpha):
@@ -85,6 +86,17 @@ def test_prelu(test_case):
             _test_prelu_ndims(test_case, *arg)
             _test_prelu_grad(test_case, *arg)
 
+    @unittest.skip("prelu has bug")
+    @autotest()
+    def test_prelu_module_with_random_data(test_case):
+        m = torch.nn.PReLU(num_parameters=random().to(int), init=random().to(float))
+        m.train(random())
+        device = random_device()
+        m.to(device)
+        x = random_pytorch_tensor().to(device)
+        y = m(x)
+        return y
+
 
 if __name__ == "__main__":
     unittest.main()

oneflow/python/test/modules/test_reshape.py

@@ -20,6 +20,7 @@
 
 import oneflow as flow
 from test_util import GenArgList
+from automated_test_util import *
 
 
 def _test_reshape(test_case, device):
@@ -42,16 +43,6 @@ def _test_reshape_tuple(test_case, device):
     test_case.assertTrue(np.array_equal(of_shape, np_shape))
 
 
-def _test_tensor_reshape(test_case, device):
-    x = np.array(
-        [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
-    ).astype(np.float32)
-    input = flow.Tensor(x, device=flow.device(device))
-    of_shape = input.reshape(shape=[2, 2, 2, -1]).numpy().shape
-    np_shape = (2, 2, 2, 2)
-    test_case.assertTrue(np.array_equal(of_shape, np_shape))
-
-
 def _test_reshape_backward(test_case, device):
     x = np.array(
         [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
@@ -77,13 +68,19 @@ def test_reshape(test_case):
         arg_dict["test_fun"] = [
             _test_reshape,
             _test_reshape_tuple,
-            _test_tensor_reshape,
             _test_reshape_backward,
         ]
         arg_dict["device"] = ["cpu", "cuda"]
         for arg in GenArgList(arg_dict):
             arg[0](test_case, *arg[1:])
 
+    @autotest()
+    def test_reshape_flow_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor(ndim=4).to(device)
+        y = torch.reshape(x, shape=(-1,))
+        return y
+
 
 if __name__ == "__main__":
     unittest.main()

oneflow/python/test/modules/test_squeeze.py

@@ -99,25 +99,12 @@ def test_squeeze(test_case):
         for arg in GenArgList(arg_dict):
             arg[0](test_case, *arg[1:])
 
+    @autotest()
     def test_flow_squeeze_with_random_data(test_case):
-        for device in ["cpu", "cuda"]:
-            test_flow_against_pytorch(
-                test_case,
-                "squeeze",
-                extra_annotations={"dim": int,},
-                extra_generators={"dim": random(0, 6)},
-                device=device,
-            )
-
-    def test_flow_tensor_squeeze_with_random_data(test_case):
-        for device in ["cpu", "cuda"]:
-            test_tensor_against_pytorch(
-                test_case,
-                "squeeze",
-                extra_annotations={"dim": int},
-                extra_generators={"dim": random(0, 6)},
-                device=device,
-            )
+        device = random_device()
+        x = random_pytorch_tensor().to(device)
+        y = torch.squeeze(x, random(1, 3).to(int))
+        return y
 
 
 if __name__ == "__main__":

oneflow/python/test/modules/test_transpose.py

@@ -20,6 +20,7 @@
 
 import oneflow as flow
 from test_util import GenArgList
+from automated_test_util import *
 
 
 def _test_transpose(test_case, device):
@@ -88,6 +89,13 @@ def test_transpose(test_case):
         for arg in GenArgList(arg_dict):
             arg[0](test_case, *arg[1:])
 
+    @autotest()
+    def test_transpose_flow_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor(ndim=4).to(device)
+        y = torch.transpose(x, dim0=random(1, 3).to(int), dim1=random(1, 3).to(int))
+        return y
+
 
 if __name__ == "__main__":
     unittest.main()

oneflow/python/test/modules/test_unsqueeze.py

@@ -20,6 +20,7 @@
 
 import oneflow as flow
 from test_util import GenArgList
+from automated_test_util import *
 
 
 def _test_unsqueeze(test_case, device):
@@ -69,6 +70,13 @@ def test_unsqueeze(test_case):
         for arg in GenArgList(arg_dict):
             arg[0](test_case, *arg[1:])
 
+    @autotest()
+    def test_flow_unsqueeze_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor().to(device)
+        y = torch.unsqueeze(x, random(1, 3).to(int))
+        return y
+
 
 if __name__ == "__main__":
     unittest.main()

oneflow/python/test/tensor/test_tensor.py

@@ -580,6 +580,55 @@ def test_tensor_round(test_case):
             np.allclose(of_input.grad.numpy(), np.zeros(shape), 1e-4, 1e-4)
         )
 
+    def _test_tensor_reshape(test_case):
+        x = np.array(
+            [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
+        ).astype(np.float32)
+        input = flow.Tensor(x)
+        of_shape = input.reshape(shape=[2, 2, 2, -1]).numpy().shape
+        np_shape = (2, 2, 2, 2)
+        test_case.assertTrue(np.array_equal(of_shape, np_shape))
+
+    @autotest()
+    def test_reshape_tensor_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor(ndim=4).to(device)
+        y = x.reshape(shape=(-1,))
+        return y
+
+    @autotest()
+    def test_tensor_squeeze_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor().to(device)
+        y = x.squeeze(random().to(int))
+        return y
+
+    @autotest()
+    def test_flow_unsqueeze_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor().to(device)
+        y = x.unsqueeze(random(1, 3).to(int))
+        return y
+
+    @autotest()
+    def test_permute_flow_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor(ndim=4).to(device)
+        y = x.permute(
+            random(0, 4).to(int),
+            random(0, 4).to(int),
+            random(0, 4).to(int),
+            random(0, 4).to(int),
+        )
+        return y
+
+    @autotest()
+    def test_transpose_tensor_with_random_data(test_case):
+        device = random_device()
+        x = random_pytorch_tensor(ndim=4).to(device)
+        y = x.transpose(dim0=random(1, 3).to(int), dim1=random(1, 3).to(int))
+        return y
+
     def test_tensor_where(test_case):
         x = flow.Tensor(
             np.array([[-0.4620, 0.3139], [0.3898, -0.7197], [0.0478, -0.1657]]),