Efficient CUDA Polynomial Preconditioned Conjugate Gradient Solver for Finite Element Computation of Elasticity Problems

Algorithm 2

SpMV kernel for the ELLPACK format.

spmv_ell_kernel  (int 𝑁, intMaxNzPerRow, int^∗𝐽𝐴,

       float^∗𝐴,float^∗𝑥,float^∗𝑦)

{

 int𝑡𝑖𝑑=𝑏𝑙𝑜𝑐𝑘𝐷𝑖𝑚.𝑥∗𝑏𝑙𝑜𝑐𝑘𝐼𝑑𝑥.𝑥+𝑡ℎ𝑟𝑒𝑎𝑑𝐼𝑑𝑥.𝑥;

 for(𝑖=𝑡𝑖𝑑;𝑖<𝑁;𝑖+=𝑏𝑙𝑜𝑐𝑘𝐷𝑖𝑚.𝑥∗𝑔𝑟𝑖𝑑𝐷𝑖𝑚.𝑥){

  for(𝑘=0;𝑘< MaxNzPerRow; 𝑘 ++)

   𝑦[𝑖]+=𝐴[𝑖+𝑁×𝑘]×𝑥[𝐽𝐴[𝑖+𝑁×𝑘]];

         }

}