1c96caaccSSatish Balay! 2c96caaccSSatish Balay! 3c96caaccSSatish Balay! Fortran kernel for sparse matrix-vector product in the AIJ matrix format 4c96caaccSSatish Balay! 5c96caaccSSatish Balay#include <petsc/finclude/petscsys.h> 6c96caaccSSatish Balay! 70ccf82acSMartin Diehlpure subroutine FortranMultTransposeAddAIJ(n,x,ii,jj,a,y) 8*fe66ebccSMartin Diehl use, intrinsic :: ISO_C_binding 90ccf82acSMartin Diehl implicit none (type, external) 100ccf82acSMartin Diehl PetscScalar, intent(in) :: x(0:*),a(0:*) 110ccf82acSMartin Diehl PetscScalar, intent(inout) :: y(0:*) 120ccf82acSMartin Diehl PetscInt, intent(in) :: n,ii(*),jj(0:*) 13c96caaccSSatish Balay 14d66e387eSMartin Diehl PetscInt :: i,jstart,jend 15c96caaccSSatish Balay 16c96caaccSSatish Balay jend = ii(1) 170113e719SMartin Diehl do i=1,n 18c96caaccSSatish Balay jstart = jend 19c96caaccSSatish Balay jend = ii(i+1) 20d66e387eSMartin Diehl y(jj(jstart:jend-1)) = y(jj(jstart:jend-1)) + x(i-1)*a(jstart:jend-1) 210113e719SMartin Diehl end do 220113e719SMartin Diehlend subroutine FortranMultTransposeAddAIJ 23c96caaccSSatish Balay 240ccf82acSMartin Diehlpure subroutine FortranMultAIJ(n,x,ii,jj,a,y) 25*fe66ebccSMartin Diehl use, intrinsic :: ISO_C_binding 260ccf82acSMartin Diehl implicit none (type, external) 270ccf82acSMartin Diehl PetscScalar, intent(in) :: x(0:*),a(0:*) 280ccf82acSMartin Diehl PetscScalar, intent(inout) :: y(*) 290ccf82acSMartin Diehl PetscInt, intent(in) :: n,ii(*),jj(0:*) 30c96caaccSSatish Balay 31d66e387eSMartin Diehl PetscInt :: i,jstart,jend 32c96caaccSSatish Balay 335161815cSJunchao Zhang#ifdef PETSC_USE_OPENMP_KERNELS 34ff45ff59SMartin Diehl !omp parallel do private(jstart,jend) 355161815cSJunchao Zhang#endif 360113e719SMartin Diehl do i=1,n 375161815cSJunchao Zhang jstart = ii(i) 38c96caaccSSatish Balay jend = ii(i+1) 39d66e387eSMartin Diehl y(i) = sum(a(jstart:jend-1)*x(jj(jstart:jend-1))) 400113e719SMartin Diehl end do 410113e719SMartin Diehlend subroutine FortranMultAIJ 42