Comment faire pour disperser un tableau numpy en python en utilisant comm.Scatterv

Question

Je suis tring pour écrire un code basé sur MPI pour faire un certain calcul en utilisant python et MPI4py. Cependant, à la suite de l'exemple, je ne peux pas disperser un vecteur numpy dans les cœurs. Voici le code et les erreurs, est-ce que quelqu'un peut m'aider? Merci.

import numpy as np 
from mpi4py import MPI 

comm = MPI.COMM_WORLD 
size = comm.Get_size() 
rank = comm.Get_rank() 
n = 6 

if rank == 0: 
    d1 = np.arange(1, n+1) 
    split = np.array_split(d1, size) 
    split_size = [len(split[i]) for i in range(len(split))] 
    split_disp = np.insert(np.cumsum(split_size), 0, 0)[0:-1] 

else: 
#Create variables on other cores 
    d1 = None 
    split = None 
    split_size = None 
    split_disp = None 

split_size = comm.bcast(split_size, root = 0) 
split_disp = comm.bcast(split_disp, root = 0) 
d1_local = np.zeros(split_size[rank]) 
comm.Scatterv([d1, split_size, split_disp, MPI.DOUBLE], d1_local, root=0) 
print('rank ', rank, ': ', d1_local) 

Et le résultat d'erreur est:

rank 2 : [ 2.47032823e-323] 
rank 3 : [ 2.96439388e-323] 
rank 0 : [ 4.94065646e-324 9.88131292e-324] 
rank 1 : [ 1.48219694e-323 1.97626258e-323] 

Merci.

Answer 1

Le type de données est incorrect. Je devrais spécifier le type du tableau:

d1 = np.arange(1, n+1, dtype='float64')