diff --git a/python/paddle/fluid/tests/unittests/test_eigh_op.py b/python/paddle/fluid/tests/unittests/test_eigh_op.py
index 9c9cd883313a2425b11b282ee9039b952a0852f7..2abbcc98a6b7e33b0e3c99e971978dde6a19fc13 100644
--- a/python/paddle/fluid/tests/unittests/test_eigh_op.py
+++ b/python/paddle/fluid/tests/unittests/test_eigh_op.py
@@ -19,8 +19,6 @@ import numpy as np
 import paddle
 from op_test import OpTest
 from gradient_checker import grad_check
-from paddle.fluid.framework import _enable_legacy_dygraph
-_enable_legacy_dygraph()
 
 
 def valid_eigh_result(A, eigh_value, eigh_vector, uplo):
diff --git a/python/paddle/utils/code_gen/backward.yaml b/python/paddle/utils/code_gen/backward.yaml
index c875162dcd4e037792398d1eac9f4c70f7badebb..1e58c19728adcc720615ae362c0f2732b8e75fdb 100644
--- a/python/paddle/utils/code_gen/backward.yaml
+++ b/python/paddle/utils/code_gen/backward.yaml
@@ -471,6 +471,9 @@
     param : [out_v]
   kernel :
     func : eigh_grad
+    data_type : out_v
+  data_transform:
+    skip_transform : out_w, out_w_grad
 
 - backward_api : elementwise_pow_grad
   forward : elementwise_pow(Tensor x, Tensor y) -> Tensor(out)