Basic code for SpMV on GPU

Open discussion for MAGMA library (Matrix Algebra on GPU and Multicore Architectures)

Basic code for SpMV on GPU

Postby akamo » Tue Feb 14, 2017 2:35 am

Hello,
I am a new user of Magma. I am looking for a basic code example for running the SpMV (Sparse Matrix Vector Multiplication), lets say using the CSR format, on the GPU side. The examples I found in the downloaded package were not very helpful because I just switched to Magma and the code is all new for me.
Thanks,
akamo
 
Posts: 2
Joined: Tue Feb 14, 2017 2:26 am

Re: Basic code for SpMV on GPU

Postby hartwig anzt » Tue Feb 14, 2017 1:38 pm

Dear Akrem Benatia,

assuming you have your matrix of size n in the CSR-array *row *col *val on the CPU, you can pass this matrix to MAGMA-sparse via

magma_d_matrix A={Magma_CSR},
magma_zcsrset( n, n, row, col, val &A, queue );

Then, you may want to initialize vectors on the GPU and copy the system to the device:

magma_d_matrix dA={Magma_CSR}, dx={Magma_DENSE}, dy={Magma_DENSE};
magma_dvinit( &dx, Magma_DEV, n, 1, 1.0, queue );
magma_dvinit( &dx, Magma_DEV, n, 1, 0.0, queue );
magma_dmtransfer( A, &dA, Magma_CPU, Magma_DEV, queue );

Now you can run your CSR-SpMV y=Ax:

magma_d_spmv( 1.0, A, x, 0.0, y, queue );


In case you want to use a different format, e.g. SELLP,, you have to convert the matrix first:

magma_d_matrix B={Magma_CSR}, dA={Magma_CSR}, dx={Magma_DENSE}, dy={Magma_DENSE};
magma_dvinit( &dx, Magma_DEV, n, 1, 1.0, queue );
magma_dvinit( &dx, Magma_DEV, n, 1, 0.0, queue );

// you can modify parameters in SELLP:
B.blocksize = 32; // as example: row-blocks of 32
B.alignment = 1; // as example: 1 thread per row, multiple are possible
magma_dmconvert( A, &B, Magma_CSR, Magma_SELLP, queue );
magma_dmtransfer( B, &dA, Magma_CPU, Magma_DEV, queue );

Now you can run your SELLP-SpMV y=Ax:

magma_d_spmv( 1.0, A, x, 0.0, y, queue );


Hope this helps! Please let me know if you have further questions.

Hartwig
hartwig anzt
 
Posts: 76
Joined: Tue Sep 02, 2014 5:44 pm

Re: Basic code for SpMV on GPU

Postby akamo » Thu Feb 16, 2017 2:21 am

Dear Mr. Hartwig
Thank you very much, that was very helpful.

Akrem
akamo
 
Posts: 2
Joined: Tue Feb 14, 2017 2:26 am


Return to User discussion

Who is online

Users browsing this forum: Google [Bot] and 2 guests