fix pylayer_memleak (#42501)

2d29d833 · wanghuancoder · GitHub · f315489d · 2d29d833
显示空白变更内容
内联并排

Showing with 4 addition and 1 deletion

paddle/fluid/eager/pylayer/py_layer_node.h paddle/fluid/eager/pylayer/py_layer_node.h +4 -1

未找到文件。
--- a/paddle/fluid/eager/pylayer/py_layer_node.h
+++ b/paddle/fluid/eager/pylayer/py_layer_node.h
@@ -32,7 +32,10 @@ class GradNodePyLayer : public GradNodeBase {
    ctx_ = ctx;
  }

-  ~GradNodePyLayer() override { Py_DECREF(ctx_); };
+  ~GradNodePyLayer() override {
+    Py_DECREF(ctx_);
+    Py_XDECREF(outputs_);
+  };

  virtual paddle::small_vector<std::vector<paddle::experimental::Tensor>,
                               kSlotSmallVectorSize>