浏览代码

Fix memleak when count is 0

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

+ 6 - 0
mpi/src/starpu_mpi_collective.c

@@ -76,6 +76,12 @@ int starpu_mpi_scatter_detached(starpu_data_handle_t *data_handles, int count, i
 				}
 			}
 		}
+
+		if (!callback_arg->count)
+		{
+			free(callback_arg);
+			return 0;
+		}
 	}
 
 	for(x = 0; x < count ; x++)