次元 (N,M) の C 順序行列があります。
mat = np.random.randn(N, M)
そのうち、永続的な MPI リクエストを介して別のノードに列を送信したいと考えています。ただし、 を使用するmpi4py
と、
sreq = MPI.COMM_WORLD.Send_Init((mat[:,idx], MPI.DOUBLE), send_id, tag)
スライスが連続していないために失敗します。誰かがこれについての方法を提案できますか? 私は、型を作成するときにMPI_Type_vector
a を指定できるC を信じています。stride
どうすればこれを達成できmpi4py
ますか?