[petsc-users] Issue to set values to a matrix in parallel in python

[Thomas Saigre]

Hi,

I've been trying for a few weeks to construct a matrix from a list of 
vectors, unsuccessfully.
Here is my problem : I have a list l a petsc4py.Vec, each vector has a 
size n, and I have d vectors. I want to "cast" these vectors to a 
petsc4py.Mat Z of shape (n,d), where Z[:, i] = l[i] (using NumPy notation)

Here is the code I'm using :

import sys
from petsc4py import PETSc
n = 5
d = 10

l = []   # creation of the list of vectors
for i in range(d):
     v = PETSc.Vec().create()
     v.setSizes(n)
     v.setFromOptions()
     v.set(i)
     l.append(v)

Z = PETSc.Mat().create()
Z.setSizes([n, d])
Z.setFromOptions()
Z.setUp()
for i, v in enumerate(l):
     Z.setValues(range(n), i, v)
Z.assemble()
Z.view()    # to display the matrix in the terminal

In sequential, the result is correct :

Mat Object: 1 MPI processes
   type: seqaij
row 0: (0, 0.)  (1, 1.)  (2, 2.)  (3, 3.)  (4, 4.)  (5, 5.)  (6, 6.)  
(7, 7.)  (8, 8.)  (9, 9.)
row 1: (0, 0.)  (1, 1.)  (2, 2.)  (3, 3.)  (4, 4.)  (5, 5.)  (6, 6.)  
(7, 7.)  (8, 8.)  (9, 9.)
row 2: (0, 0.)  (1, 1.)  (2, 2.)  (3, 3.)  (4, 4.)  (5, 5.)  (6, 6.)  
(7, 7.)  (8, 8.)  (9, 9.)
row 3: (0, 0.)  (1, 1.)  (2, 2.)  (3, 3.)  (4, 4.)  (5, 5.)  (6, 6.)  
(7, 7.)  (8, 8.)  (9, 9.)
row 4: (0, 0.)  (1, 1.)  (2, 2.)  (3, 3.)  (4, 4.)  (5, 5.)  (6, 6.)  
(7, 7.)  (8, 8.)  (9, 9.)

  but when I run it using the command mpirun -np 2 python3 file.py, I 
get the following error, about incompatible array sizes (I did not 
manage to understand what ni, nj and nv correspond to...)

Traceback (most recent call last):
   File "/home/Documents/code/tests/file.py", line 31, in <module>
     Z.setValues(list(range(n)), i, v)
   File "PETSc/Mat.pyx", line 888, in petsc4py.PETSc.Mat.setValues
   File "PETSc/petscmat.pxi", line 828, in petsc4py.PETSc.matsetvalues
ValueError: incompatible array sizes: ni=5, nj=1, nv=3
Traceback (most recent call last):
   File "/home/saigre/Documents/code/tests/t2.py", line 31, in <module>
     Z.setValues(list(range(n)), i, v)
   File "PETSc/Mat.pyx", line 888, in petsc4py.PETSc.Mat.setValues
   File "PETSc/petscmat.pxi", line 828, in petsc4py.PETSc.matsetvalues
ValueError: incompatible array sizes: ni=5, nj=1, nv=2

Two weeks ago, I made a post on stack overflow 
(https://stackoverflow.com/questions/73124230/convert-a-list-of-vector-to-a-matrix-with-petsc4py). 
I tried using the apt packages, and I also compiled from the sources, 
but I get the same error.

I someone has an idea how to succeed in it, I'm all ears !

Thanks,

Thomas