diff --git a/paddle/fluid/framework/ir/node.h b/paddle/fluid/framework/ir/node.h index e41e032d14210b925166ea445be8c62298648697..1b7364858dcbbbcfbfcb906723c0045aa0af6ab8 100644 --- a/paddle/fluid/framework/ir/node.h +++ b/paddle/fluid/framework/ir/node.h @@ -27,7 +27,11 @@ namespace ir { // Node should normally created by Graph::CreateXXXNode(). class Node { public: - virtual ~Node() {} + virtual ~Node() { + if (!wrapper_.empty()) { + wrapper_deleter_(); + } + } enum class Type { kOperation, kVariable }; static constexpr char kControlDepVarName[] = "__control_var";