fixed memory leak

Jiaming Guo
1 parent 91589064
Showing 1 changed file with 16 additions and 5 deletions Show diff stats
stim/biomodels/flow.h
@@ -118,6 +118,22 @@ namespace stim {
 			Q.resize(n_e);
 		}
+		void reset(unsigned n_v) {
+			
+			for (unsigned i = 0; i < n_v; i++) {
+				for (unsigned j = 0; j < n_v; j++) {
+					C[i][j] = 0;
+				}
+			}
+		}
+
+		void clear(unsigned n_v) {
+			
+			for (unsigned i = 0; i < n_v; i++)
+				delete[] C[i];
+			delete[] C;
+		}
+
 		/// Calculate the inverse of A and store the result in C
 		void inversion(T** A, int order, T* C) {
@@ -128,7 +144,6 @@ namespace stim {
 			for (unsigned i = 0; i < order; i++)
 				for (unsigned j = 0; j < order; j++)
 					Aflat[i * order + j] = A[i][j];
-			//T* Cflat = C;// (T*)malloc(order * order * sizeof(T));
 			// create device pointer
 			T* d_Aflat;		// flat original matrix
@@ -178,12 +193,8 @@ namespace stim {
 			// copy inverse matrix from device to host
 			HANDLE_ERROR(cudaMemcpy(C, d_Cflat, order * order * sizeof(T), cudaMemcpyDeviceToHost));
-			//for(unsigned i = 0; i < order; i++)
-			//	memcpy(C[i], &Cflat[i*order], order * sizeof(T*));
-
 			// clear up
 			free(Aflat);
-			//free(Cflat);
 			HANDLE_ERROR(cudaFree(d_Aflat));
 			HANDLE_ERROR(cudaFree(d_Cflat));
 			HANDLE_ERROR(cudaFree(d_A));