fix leak

[simgrid.git] / src / smpi / colls / smpi_default_selector.cpp

diff --git a/src/smpi/colls/smpi_default_selector.cpp b/src/smpi/colls/smpi_default_selector.cpp
index febd2e3..081e889 100644 (file)
--- a/src/smpi/colls/smpi_default_selector.cpp
+++ b/src/smpi/colls/smpi_default_selector.cpp
@@ -39,10 +39,20 @@ int reduce_scatter__default(const void *sendbuf, void *recvbuf, const int *recvc
   int size = comm->size();
   int count = 0;
   int* displs = new int[size];
+  int regular=1;
   for (int i = 0; i < size; i++) {
+    if(recvcounts[i]!=recvcounts[0]){
+      regular=0;
+      continue;
+    }
     displs[i] = count;
     count += recvcounts[i];
   }
+  if(not regular){
+    delete[] displs;
+    return reduce_scatter__ompi(sendbuf, recvbuf, recvcounts, datatype, op, comm);
+  }
+
   unsigned char* tmpbuf = smpi_get_tmp_sendbuffer(count * datatype->get_extent());
 
   int ret = reduce__default(sendbuf, tmpbuf, count, datatype, op, 0, comm);