浏览代码

mlr: fixing memory leak when dgels doesnt work properly

Luka Stanisic 8 年之前
父节点
当前提交
fa46554522
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      src/core/perfmodel/multiple_regression.c

+ 4 - 0
src/core/perfmodel/multiple_regression.c

@@ -308,7 +308,11 @@ int _starpu_multiple_regression(struct starpu_perfmodel_history_list *ptr, doubl
 #ifdef STARPU_MLR_MODEL
 		/* Computing coefficients using multiple linear regression */
 		if(dgels_multiple_reg_coeff(mpar, my, n, ncoeff, nparameters, coeff, combinations))
+		{
+			free(mpar);
+			free(my);	
 			return 1;
+		}
 		/* Basic validation of the model accuracy */
 		validate(coeff, ncoeff, codelet_name);
 #else