Merge pull request #1247 from saatviks/saatviks_tfmemleakfix

pervazea · web-flow · commit 5b16e38eb477 · 2018-04-02T13:05:43.000-04:00
Tensorflow memory leak fix
diff --git a/src/brain/tf_session_entity/tf_session_entity.cpp b/src/brain/tf_session_entity/tf_session_entity.cpp
@@ -31,8 +31,11 @@ TFSE_TYPE::TfSessionEntity() {
 
 TFSE_TEMPLATE_ARGUMENTS
 TFSE_TYPE::~TfSessionEntity() {
-  TF_DeleteStatus(status_);
+  TF_CloseSession(session_, status_);
+  TF_DeleteSession(session_, status_);
   TF_DeleteGraph(graph_);
+  TF_DeleteStatus(status_);
+  TF_DeleteSessionOptions(session_options_);
 }
 
 /*
diff --git a/test/leak_suppr.txt b/test/leak_suppr.txt
@@ -3,3 +3,5 @@
 # see #1194
 # 
 leak:src_backend_utils_mmgr_mcxt.c
+# suppress the tensorflow background thread problem(#1232)
+leak:tf_session_entity.cpp

Original file line number	Diff line number	Diff line change
`@@ -3,3 +3,5 @@`
`3`	`3`	`# see #1194`
`4`	`4`	`#`
`5`	`5`	`leak:src_backend_utils_mmgr_mcxt.c`
	`6`	`+# suppress the tensorflow background thread problem(#1232)`
	`7`	`+leak:tf_session_entity.cpp`