diff --git a/python/paddle/fluid/tests/unittests/test_cumsum_op.py b/python/paddle/fluid/tests/unittests/test_cumsum_op.py index 0ac4f728cfdd8a4211b001543092b6ddb773200c..68bb8b17c5d85b52a924bc59f4c1fb4eea4aa776 100644 --- a/python/paddle/fluid/tests/unittests/test_cumsum_op.py +++ b/python/paddle/fluid/tests/unittests/test_cumsum_op.py @@ -27,10 +27,10 @@ class TestSumOp1(OpTest): self.outputs = {'Out': self.inputs['X'].cumsum(axis=2)} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp2(OpTest): @@ -45,10 +45,10 @@ class TestSumOp2(OpTest): } def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp3(OpTest): @@ -59,10 +59,10 @@ class TestSumOp3(OpTest): self.outputs = {'Out': self.inputs['X'].cumsum(axis=1)} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp4(OpTest): @@ -73,10 +73,10 @@ class TestSumOp4(OpTest): self.outputs = {'Out': self.inputs['X'].cumsum(axis=0)} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp5(OpTest): @@ -86,10 +86,10 @@ class TestSumOp5(OpTest): self.outputs = {'Out': self.inputs['X'].cumsum(axis=1)} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp7(OpTest): @@ -99,10 +99,10 @@ class TestSumOp7(OpTest): self.outputs = {'Out': self.inputs['X'].cumsum(axis=0)} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') class TestSumOp8(OpTest): @@ -119,10 +119,10 @@ class TestSumOp8(OpTest): } def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', check_dygraph=False) + self.check_grad(['X'], 'Out') if __name__ == '__main__': diff --git a/python/paddle/fluid/tests/unittests/test_fill_op.py b/python/paddle/fluid/tests/unittests/test_fill_op.py index 36b56ab92801c270552c100d1b68d48678e3320d..7c8587dc400206628c10a7ed35e401da953c8e10 100644 --- a/python/paddle/fluid/tests/unittests/test_fill_op.py +++ b/python/paddle/fluid/tests/unittests/test_fill_op.py @@ -35,7 +35,7 @@ class TestFillOp1(OpTest): self.outputs = {'Out': val.astype('float64')} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() class TestFillOp2(OpTest): @@ -52,7 +52,7 @@ class TestFillOp2(OpTest): self.outputs = {'Out': val.astype('float64')} def test_check_output(self): - self.check_output(check_dygraph=False) + self.check_output() class TestFillOp3(unittest.TestCase):