From 2d29d833c3f0684a91a545e4284de97ac875808f Mon Sep 17 00:00:00 2001 From: wanghuancoder Date: Fri, 6 May 2022 09:30:44 +0800 Subject: [PATCH] fix pylayer_memleak (#42501) --- paddle/fluid/eager/pylayer/py_layer_node.h | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/eager/pylayer/py_layer_node.h b/paddle/fluid/eager/pylayer/py_layer_node.h index affed77019..b477d7a9ad 100644 --- 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, kSlotSmallVectorSize> -- GitLab