Memory leak in the relay interpreter (#3448)

9b3e83f6 · hlu1 · Tianqi Chen · b05dc12f · 9b3e83f6
Commit 9b3e83f6 authored Jun 27, 2019 by hlu1 Committed by Tianqi Chen Jun 27, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

src/relay/backend/interpreter.cc
+3 -1

No files found.
--- a/src/relay/backend/interpreter.cc
+++ b/src/relay/backend/interpreter.cc
@@ -299,7 +299,9 @@ class Interpreter :
    auto closure = ClosureNode::make(captured_mod, func);
    auto mut_closure =
        static_cast<ClosureNode*>(const_cast<Node*>(closure.get()));
-    mut_closure->env.Set(letrec_name, closure);
+    if (letrec_name.defined()) {
+      mut_closure->env.Set(letrec_name, closure);
+    }
    return std::move(closure);
  }