From 1c908df681943623fdded59d52a54b376fb016ec Mon Sep 17 00:00:00 2001
From: Helin Wang <ustc.harry@gmail.com>
Date: Tue, 16 May 2017 16:36:33 -0400
Subject: [PATCH] fix test memory leak

---
 paddle/go/pserver/lib/client/test/main.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/paddle/go/pserver/lib/client/test/main.c b/paddle/go/pserver/lib/client/test/main.c
index 6fec3a7a83a..14d4522eac0 100644
--- a/paddle/go/pserver/lib/client/test/main.c
+++ b/paddle/go/pserver/lib/client/test/main.c
@@ -8,6 +8,20 @@ void panic() {
   *(void*)0;
 }
 
+void releaseParam(paddle_parameter* param) {
+  if (param != NULL) {
+    if (param->name != NULL) {
+      free(param->name);
+    }
+
+    if (param->content != NULL) {
+      free(param->content);
+    }
+
+    free(param);
+  }
+}
+
 int main() {
   char addr[] = "localhost:3000";
   client c = paddle_new_pserver_client(addr);
@@ -51,6 +65,9 @@ int main() {
     panic();
   }
 
+  releaseParam(params[0]);
+  releaseParam(params[1]);
+
   if (!paddle_save_model(c, "/tmp/")) {
     panic();
   }
-- 
GitLab