Changeset 691 for codes/icosagcm

Timestamp:

05/03/18 17:26:18 (6 years ago)

Author:

dubos

Message:

devel/Python : bugfix, on Irene all2all corrupts its argument ?

File:

• codes/icosagcm/devel/Python/dynamico/parallel.py (modified) (1 diff)

codes/icosagcm/devel/Python/dynamico/parallel.py

@@ -36 +36 @@
         index_list, self.order = ordered_list(index_list)
         comm, vtxdist = dim.comm, dim.vtxdist
+        self.comm, self.n, self.vtxdist = comm, len(index_list), vtxdist
         list_local = [ [ j-vtxdist[i] for j in index_list if j>=vtxdist[i] and j<vtxdist[i+1]] 
                       for i in range(len(vtxdist)-1) ]
-        list_send = comm.alltoall(list_local)
-        self.comm, self.n, self.list_n = comm, len(index_list), [len(i) for i in list_local]
-        self.list_local, self.list_send, self.dict = list_local, list_send, inverse_list(index_list)
-        self.vtxdist=vtxdist
+        self.list_local, self.list_n = list_local, map(len, list_local)
+        list_send = comm.alltoall(list_local) # may modify list_local !!
+        self.list_send, self.dict = list_send, inverse_list(index_list)
     def get(self, get_data, put_data):
         data_send = [get_data(indices) for indices in self.list_send]