浏览代码

Fix memealk on error

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

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

@@ -236,6 +236,9 @@ int dgels_multiple_reg_coeff(double *mpar, double *my, unsigned long nn, unsigne
 	if( info != 0 )
 	{
 		_STARPU_DISP("Warning: Problems when executing dgels_ function. It seems like the diagonal element %ld is zero.\n Multiple linear regression model will not be written into perfmodel file.\n", info);
+		free(X);
+		free(Y);
+		free(work);
 		return 1;
 	}