fix memory leak in trainBinarySVM

Xtra-Computing · Nov 8, 2016 · 2949e3e · 2949e3e
1 parent bdb4348
commit 2949e3e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/mascot/svmMain.cu b/mascot/svmMain.cu
@@ -54,5 +54,5 @@ int main(int argc, char **argv)
 
     }
 
-        return 0;
+    return 0;
 }
diff --git a/mascot/svmModel.cu b/mascot/svmModel.cu
@@ -1,6 +1,8 @@
-//
-// Created by ss on 16-11-2.
-//
+/*
+   * @author: created by ss on 16-11-2.
+   * @brief: multi-class svm training, prediction, svm with probability output 
+   *
+*/
 
 #include <cstdio>
 #include "svmModel.h"

diff --git a/mascot/trainingFunction.cu b/mascot/trainingFunction.cu
@@ -32,8 +32,6 @@
 using std::cout;
 using std::endl;
 
-void trainingByGPU(vector<vector<float_point> > &v_v_DocVector, data_info &SDataInfo, SVMParam &param);
-
 svmModel trainSVM(SVMParam &param, string strTrainingFileName, int nNumofFeature) {
 
     vector<vector<float_point> > v_v_DocVector;
@@ -169,6 +167,20 @@ svm_model trainBinarySVM(svmProblem &problem, const SVMParam &param) {
     }
 
     model.nDimension = problem.getNumOfFeatures();
+
+    //free device memory
+    checkCudaErrors(cudaFree(pfDevAlphaSubset));
+    checkCudaErrors(cudaFree(pnDevLabelSubset));
+    checkCudaErrors(cudaFree(pfDevYiGValueSubset));
+
+    //release pinned memory
+    cudaFreeHost(DeviceHessian::m_pfHessianRowsInHostMem);
+
+    //release host memory
+    delete[] DeviceHessian::m_pfHessianDiag;
+    delete[] pfAlphaAll;
+    delete[] pfYiGValueAll;
+
     return model;
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -54,5 +54,5 @@ int main(int argc, char **argv) @@
         }
-            return 0;
+        return 0;
     }