Python PETSc Example BP1 #1806
Python PETSc Example BP1 #1806
Conversation
examples/python/bpsraw.py
Outdated
| matrix = PETSc.Mat().createAIJ([num_dofs, num_dofs], comm=PETSc.COMM_WORLD) | ||
| matrix.setFromOptions() | ||
| matrix.setUp() | ||
| for i in range(num_dofs): | ||
| e_vec = ceed.Vector(num_dofs) | ||
| r_vec = ceed.Vector(num_dofs) | ||
| e_vec.set_value(0.0) | ||
| r_vec.set_value(0.0) | ||
| with e_vec.array_write() as arr: | ||
| arr[i] = 1.0 | ||
| op.apply(e_vec, r_vec) | ||
| with r_vec.array_read() as r_arr: | ||
| matrix.setValues(range(num_dofs), [i], r_arr) | ||
| matrix.assemble() |
There was a problem hiding this comment.
You'll want a MatShell, not a MatAIJ
There was a problem hiding this comment.
Apologies for leaving this a bit stale. Addressing this led to a lot of other changes. I had trouble setting the matrix type to "shell" and applying context, kept getting segmentation faults. So, instead of using the "shell" type, I used the "python" type and set the context through a python class. Pulled that implementation from the petsc4py 2DPoisson example (about half way down) --> link. I'll tidy up what I have a submit an update soon.
|
@tylercollins5737 , thank you for your work. Are you able to address the reviewer's comments? |
|
@valeriabarra I was able to get it working using the "python" matrix type, MatAIJ was not the best. I believe the "python" type and MatCreateShell can achieve the same thing. |
Hello!
This is a Python implementation of bpsraw.c for BP1 and was initially discussed in #1774 . I'm aiming to recreate the benchmark behavior using libceed and petsc4py. I appreciate any feedback or indications I'm headed in the right direction!
It does run, but this is a work in progress. Remaining tasks: