patch_tensor_method_func, test=develop (#38761)

PaddlePaddle · Jan 7, 2022 · c8fbd3c · c8fbd3c
1 parent 769e5bc
commit c8fbd3c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py
@@ -90,6 +90,8 @@ def _test_eager_guard(tracer=None):
     if not _already_patch_eager_tensor:
         from .dygraph.varbase_patch_methods import monkey_patch_varbase
         monkey_patch_varbase()
+        from .dygraph import monkey_patch_math_varbase
+        monkey_patch_math_varbase()
         _already_patch_eager_tensor = True
     if tracer is None:
         core._set_eager_tracer(_dygraph_tracer_)