浏览代码

Fix memleak when count is 0

Samuel Thibault 9 年之前
父节点
当前提交
256bc993b3
共有 1 个文件被更改,包括 6 次插入0 次删除
  1. 6 0
      mpi/src/starpu_mpi_collective.c

+ 6 - 0
mpi/src/starpu_mpi_collective.c

@@ -140,6 +140,12 @@ int starpu_mpi_gather_detached(starpu_data_handle_t *data_handles, int count, in
 		}
 	}
 
+	if (!callback_arg->count)
+	{
+		free(callback_arg);
+		return 0;
+	}
+
 	for(x = 0; x < count ; x++)
 	{
 		if (data_handles[x])