未验证 提交 cf799525 编写于 作者: 姜永久 提交者: GitHub

rm autograd func dynamic eager tests (#48788)

上级 a85dedf9
...@@ -24,7 +24,6 @@ from utils import matmul, mul, nested, o2, reduce, reduce_dim ...@@ -24,7 +24,6 @@ from utils import matmul, mul, nested, o2, reduce, reduce_dim
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.nn.functional as F import paddle.nn.functional as F
from paddle.fluid.framework import _test_eager_guard
from paddle.incubate.autograd.utils import as_tensors from paddle.incubate.autograd.utils import as_tensors
...@@ -201,14 +200,6 @@ class TestVJP(TestAutogradFunctional): ...@@ -201,14 +200,6 @@ class TestVJP(TestAutogradFunctional):
self.check_results(ref_result, aliased_result) self.check_results(ref_result, aliased_result)
def test_all_cases(self): def test_all_cases(self):
with _test_eager_guard():
self.func_vjp_i1o1()
self.func_vjp_i2o1()
self.func_vjp_i2o2()
self.func_vjp_i2o2_omitting_v()
self.func_vjp_nested()
self.func_vjp_aliased_input()
self.func_vjp_i1o1() self.func_vjp_i1o1()
self.func_vjp_i2o1() self.func_vjp_i2o1()
self.func_vjp_i2o2() self.func_vjp_i2o2()
...@@ -237,17 +228,12 @@ class TestVJP(TestAutogradFunctional): ...@@ -237,17 +228,12 @@ class TestVJP(TestAutogradFunctional):
), ),
) )
class TestVJPException(unittest.TestCase): class TestVJPException(unittest.TestCase):
def func_vjp(self): def test_vjp(self):
with self.assertRaises(self.expected_exception): with self.assertRaises(self.expected_exception):
paddle.incubate.autograd.vjp( paddle.incubate.autograd.vjp(
self.fun, paddle.to_tensor(self.xs), paddle.to_tensor(self.v) self.fun, paddle.to_tensor(self.xs), paddle.to_tensor(self.v)
) )
def test_all_cases(self):
with _test_eager_guard():
self.func_vjp()
self.func_vjp()
def jac(grad_fn, f, inputs): def jac(grad_fn, f, inputs):
assert grad_fn in [ assert grad_fn in [
...@@ -324,11 +310,6 @@ class TestJVP(TestAutogradFunctional): ...@@ -324,11 +310,6 @@ class TestJVP(TestAutogradFunctional):
self.check_results(results_omitting_v, results_with_v) self.check_results(results_omitting_v, results_with_v)
def test_all_cases(self): def test_all_cases(self):
with _test_eager_guard():
self.func_jvp_i1o1()
self.func_jvp_i2o1()
self.func_jvp_i2o2()
self.func_jvp_i2o2_omitting_v()
self.func_jvp_i1o1() self.func_jvp_i1o1()
self.func_jvp_i2o1() self.func_jvp_i2o1()
self.func_jvp_i2o2() self.func_jvp_i2o2()
...@@ -372,7 +353,7 @@ class TestJacobianNoBatch(unittest.TestCase): ...@@ -372,7 +353,7 @@ class TestJacobianNoBatch(unittest.TestCase):
.get("atol") .get("atol")
) )
def func_jacobian(self): def test_jacobian(self):
xs = ( xs = (
[paddle.to_tensor(x) for x in self.xs] [paddle.to_tensor(x) for x in self.xs]
if isinstance(self.xs, typing.Sequence) if isinstance(self.xs, typing.Sequence)
...@@ -409,11 +390,6 @@ class TestJacobianNoBatch(unittest.TestCase): ...@@ -409,11 +390,6 @@ class TestJacobianNoBatch(unittest.TestCase):
) )
return utils._np_concat_matrix_sequence(jac, utils.MatrixFormat.NM) return utils._np_concat_matrix_sequence(jac, utils.MatrixFormat.NM)
def test_all_cases(self):
with _test_eager_guard():
self.func_jacobian()
self.func_jacobian()
@utils.place(config.DEVICES) @utils.place(config.DEVICES)
@utils.parameterize( @utils.parameterize(
...@@ -451,7 +427,7 @@ class TestJacobianBatchFirst(unittest.TestCase): ...@@ -451,7 +427,7 @@ class TestJacobianBatchFirst(unittest.TestCase):
.get("atol") .get("atol")
) )
def func_jacobian(self): def test_jacobian(self):
xs = ( xs = (
[paddle.to_tensor(x) for x in self.xs] [paddle.to_tensor(x) for x in self.xs]
if isinstance(self.xs, typing.Sequence) if isinstance(self.xs, typing.Sequence)
...@@ -505,11 +481,6 @@ class TestJacobianBatchFirst(unittest.TestCase): ...@@ -505,11 +481,6 @@ class TestJacobianBatchFirst(unittest.TestCase):
jac, utils.MatrixFormat.NBM, utils.MatrixFormat.BNM jac, utils.MatrixFormat.NBM, utils.MatrixFormat.BNM
) )
def test_all_cases(self):
with _test_eager_guard():
self.func_jacobian()
self.func_jacobian()
class TestHessianNoBatch(unittest.TestCase): class TestHessianNoBatch(unittest.TestCase):
@classmethod @classmethod
...@@ -607,13 +578,6 @@ class TestHessianNoBatch(unittest.TestCase): ...@@ -607,13 +578,6 @@ class TestHessianNoBatch(unittest.TestCase):
paddle.incubate.autograd.Hessian(func, paddle.ones([3])) paddle.incubate.autograd.Hessian(func, paddle.ones([3]))
def test_all_cases(self): def test_all_cases(self):
with _test_eager_guard():
self.setUpClass()
self.func_single_input()
self.func_multi_input()
self.func_allow_unused_true()
self.func_create_graph_true()
self.func_out_not_single()
self.setUpClass() self.setUpClass()
self.func_single_input() self.func_single_input()
self.func_multi_input() self.func_multi_input()
...@@ -744,13 +708,6 @@ class TestHessianBatchFirst(unittest.TestCase): ...@@ -744,13 +708,6 @@ class TestHessianBatchFirst(unittest.TestCase):
) )
def test_all_cases(self): def test_all_cases(self):
with _test_eager_guard():
self.setUpClass()
self.func_single_input()
self.func_multi_input()
self.func_allow_unused()
self.func_stop_gradient()
self.func_out_not_single()
self.setUpClass() self.setUpClass()
self.func_single_input() self.func_single_input()
self.func_multi_input() self.func_multi_input()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册