[petsc-users] Matvecs and KSPSolves with multiple vectors

Barry Smith bsmith at petsc.dev
Thu Dec 7 13:34:15 CST 2023



> On Dec 7, 2023, at 1:17 PM, Sreeram R Venkat <srvenkat at utexas.edu> wrote:
> 
> I have 2 sequential matrices M and R (both MATSEQAIJCUSPARSE of size n x n) and a vector v of size n*m. v = [v_1 , v_2 ,... , v_m] where v_i has size n. The data for v can be stored either in column-major or row-major order.  Now, I want to do 2 types of operations:
> 
> 1. Matvecs of the form M*v_i = w_i, for i = 1..m. 
> 2. KSPSolves of the form R*x_i = v_i, for i = 1..m.
> 
> From what I have read on the documentation, I can think of 2 approaches. 
> 
> 1. Get the pointer to the data in v (column-major) and use it to create a dense matrix V. Then do a MatMatMult with M*V = W, and take the data pointer of W to create the vector w. For KSPSolves, use KSPMatSolve with R and V.
> 
> 2. Create a MATMAIJ using M/R and use that for matvecs directly with the vector v. I don't know if KSPSolve with the MATMAIJ will know that it is a multiple RHS system and act accordingly.
> 
> Which would be the more efficient option?

Use 1. 
> 
> As a side-note, I am also wondering if there is a way to use row-major storage of the vector v.

No

> The reason is that this could allow for more coalesced memory access when doing matvecs.

  PETSc matrix-vector products use BLAS GMEV matrix-vector products for the computation so in theory they should already be well-optimized

> 
> Thanks,
> Sreeram

-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.mcs.anl.gov/pipermail/petsc-users/attachments/20231207/825083b1/attachment.html>


More information about the petsc-users mailing list