avoid close threadlocal lua state for singleton and cudart safety (#174)

eric-haibin-lin · Sep 21, 2016 · d0e19e7 · d0e19e7
1 parent a92d7a3
commit d0e19e7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/include/dmlc/lua.h b/include/dmlc/lua.h
@@ -455,7 +455,10 @@ inline LuaState::LuaState() {
 }
 
 inline LuaState::~LuaState() {
-  lua_close(L_);
+  if (option_ != kThreadLocal && L_ != nullptr) {
+    // never close threadlocal, for save destruction.
+    lua_close(L_);
+  }
 }
 
 inline LuaState* LuaState::Create_(Option opt) {