Memory leak in the relay interpreter (#3448)

apache · Jun 28, 2019 · 9b3e83f · 9b3e83f
1 parent b05dc12
commit 9b3e83f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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);
   }