You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
psblas3/base/serial/coo/zcoorws.f

55 lines
1.3 KiB
FortranFixed

SUBROUTINE ZCOORWS(TRANS,M,N,DESCRA,A,IA1,IA2,
& INFOA,ROWSUM,IERROR)
IMPLICIT NONE
C .. Scalar Arguments ..
INTEGER M,N, IERROR
CHARACTER TRANS
C .. Array Arguments ..
INTEGER IA1(*),IA2(*),INFOA(*)
CHARACTER DESCRA*11
COMPLEX*16 A(*), ROWSUM(*)
C .. Local scalars ..
INTEGER I, J, NNZ, K
DOUBLE PRECISION SUM
logical lsame
external lsame
NNZ = INFOA(1)
IF (lsame(TRANS,'N')) THEN
DO I=1, M
ROWSUM(I) = 0.0D0
ENDDO
I = 1
J = I
DO WHILE (I.LE.NNZ)
DO WHILE ((IA1(J).EQ.IA1(I)).AND.
+ (J.LE.NNZ))
J = J+1
ENDDO
SUM = 0.0
DO K = I, J-1
SUM = SUM + ABS(REAL(A(J))) + ABS(AIMAG(A(J)))
ENDDO
ROWSUM(IA1(I)) = ROWSUM(IA1(I)) + SUM
I = J
ENDDO
ELSE IF (lsame(TRANS,'T').OR.lsame(TRANS,'C')) THEN
DO J = 1, N
ROWSUM(J) = 0.0D0
ENDDO
DO I = 1, NNZ
ROWSUM(IA2(I)) = ROWSUM(IA2(I)) +
+ ABS(REAL(A(J))) + ABS(AIMAG(A(J)))
ENDDO
ELSE
ierror = -1
ENDIF
RETURN
END