axpy.f90

Go to the documentation of this file.

SUBROUTINE axpy(N1,N2,N3,localInterval,a,X,Y,Z)

  INTEGER(KIND=8), INTENT(IN)  :: N1, N2, N3, localInterval(6)
  REAL(KIND=8),    INTENT(IN)  :: a,X(n1*n2*n3)
  REAL(KIND=8),    INTENT(IN)  :: Y(n1*n2*n3)
  REAL(KIND=8),    INTENT(OUT) :: Z(n1*n2*n3)

  INTEGER(KIND=8) :: I, J, K, LI, LIK, LIJ, IS1, IE1
  INTEGER(KIND=8) :: xyPlane
  INTEGER(KIND=8) :: S1,E1,S2,E2,S3,E3

  s1 = localinterval(1)
  e1 = localinterval(2)
  s2 = localinterval(3)
  e2 = localinterval(4)
  s3 = localinterval(5)
  e3 = localinterval(6)

  yzplane = (n2*n3)
  
  DO i = s1, e1
     !LI = (I-1)
     DO j = s2, e2
        lij = (j-1)*n1 + i
        DO k = s3, e3
           lijk = (k-1)*yzplane + lij
           z(lijk) = a*x(lijk) + y(lijk)
        END DO
     END DO
  END DO
  

END SUBROUTINE axpy