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/csr/csrsv.f

177 lines
5.8 KiB
Fortran

C
C Parallel Sparse BLAS version 2.2
C (C) Copyright 2006/2007/2008
C Salvatore Filippone University of Rome Tor Vergata
C Alfredo Buttari University of Rome Tor Vergata
C
C Redistribution and use in source and binary forms, with or without
C modification, are permitted provided that the following conditions
C are met:
C 1. Redistributions of source code must retain the above copyright
C notice, this list of conditions and the following disclaimer.
C 2. Redistributions in binary form must reproduce the above copyright
C notice, this list of conditions, and the following disclaimer in the
C documentation and/or other materials provided with the distribution.
C 3. The name of the PSBLAS group or the names of its contributors may
C not be used to endorse or promote products derived from this
C software without specific written permission.
C
C THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
C ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
C TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
C PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PSBLAS GROUP OR ITS CONTRIBUTORS
C BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
C CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
C SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
C INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
C CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
C ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
C POSSIBILITY OF SUCH DAMAGE.
C
SUBROUTINE CSRSV (UPLO,TRANS,DIAG,N,AS,JA,IA,B,X)
use psb_const_mod
use psb_string_mod
complex(psb_spk_) ZERO
PARAMETER (ZERO = (0.0, 0.0))
INTEGER N
CHARACTER DIAG, TRANS, UPLO
complex(psb_spk_) AS(*), B(*), X(*)
INTEGER IA(*), JA(*)
complex(psb_spk_) ACC
INTEGER I, J, K
LOGICAL LOW, TRA, COTRA, UNI
UNI = (psb_toupper(DIAG).EQ.'U')
TRA = (psb_toupper(TRANS).EQ.'T')
COTRA = (psb_toupper(TRANS).EQ.'C')
LOW = (psb_toupper(UPLO).EQ.'L')
IF ((.NOT.TRA).AND.(.NOT.COTRA)) THEN
IF (LOW) THEN
IF (.NOT.UNI) THEN
DO 40 I = 1, N
ACC = ZERO
DO 20 J = IA(I), IA(I+1) - 2
ACC = ACC + AS(J)*X(JA(J))
20 CONTINUE
X(I) = (B(I)-ACC)/AS(IA(I+1)-1)
40 CONTINUE
ELSE IF (UNI) THEN
DO 80 I = 1, N
ACC = ZERO
DO 60 J = IA(I), IA(I+1) - 1
ACC = ACC + AS(J)*X(JA(J))
60 CONTINUE
X(I) = B(I) - ACC
80 CONTINUE
END IF
ELSE IF (.NOT.LOW) THEN
IF (.NOT.UNI) THEN
DO 120 I = N, 1, -1
ACC = ZERO
DO 100 J = IA(I) + 1, IA(I+1) - 1
ACC = ACC + AS(J)*X(JA(J))
100 CONTINUE
X(I) = (B(I)-ACC)/AS(IA(I))
120 CONTINUE
ELSE IF (UNI) THEN
DO 160 I = N, 1, -1
ACC = ZERO
DO 140 J = IA(I), IA(I+1) - 1
ACC = ACC + AS(J)*X(JA(J))
140 CONTINUE
X(I) = B(I) - ACC
160 CONTINUE
END IF
END IF
ELSE IF (TRA) THEN
DO 180 I = 1, N
X(I) = B(I)
180 CONTINUE
IF (LOW) THEN
IF (.NOT.UNI) THEN
DO 220 I = N, 1, -1
X(I) = X(I)/AS(IA(I+1)-1)
ACC = X(I)
DO 200 J = IA(I), IA(I+1) - 2
K = JA(J)
X(K) = X(K) - AS(J)*ACC
200 CONTINUE
220 CONTINUE
ELSE IF (UNI) THEN
DO 260 I = N, 1, -1
ACC = X(I)
DO 240 J = IA(I), IA(I+1) - 1
K = JA(J)
X(K) = X(K) - AS(J)*ACC
240 CONTINUE
260 CONTINUE
END IF
ELSE IF (.NOT.LOW) THEN
IF (.NOT.UNI) THEN
DO 300 I = 1, N
X(I) = X(I)/AS(IA(I))
ACC = X(I)
DO 280 J = IA(I) + 1, IA(I+1) - 1
K = JA(J)
X(K) = X(K) - AS(J)*ACC
280 CONTINUE
300 CONTINUE
ELSE IF (UNI) THEN
DO 340 I = 1, N
ACC = X(I)
DO 320 J = IA(I), IA(I+1) - 1
K = JA(J)
X(K) = X(K) - AS(J)*ACC
320 CONTINUE
340 CONTINUE
END IF
END IF
ELSE IF (COTRA) THEN
DO 580 I = 1, N
X(I) = B(I)
580 CONTINUE
IF (LOW) THEN
IF (.NOT.UNI) THEN
DO 620 I = N, 1, -1
X(I) = X(I)/CONJG(AS(IA(I+1)-1))
ACC = X(I)
DO 600 J = IA(I), IA(I+1) - 2
K = JA(J)
X(K) = X(K) - CONJG(AS(J))*ACC
600 CONTINUE
620 CONTINUE
ELSE IF (UNI) THEN
DO 660 I = N, 1, -1
ACC = X(I)
DO 640 J = IA(I), IA(I+1) - 1
K = JA(J)
X(K) = X(K) - CONJG(AS(J))*ACC
640 CONTINUE
660 CONTINUE
END IF
ELSE IF (.NOT.LOW) THEN
IF (.NOT.UNI) THEN
DO 700 I = 1, N
X(I) = X(I)/CONJG(AS(IA(I)))
ACC = X(I)
DO 680 J = IA(I) + 1, IA(I+1) - 1
K = JA(J)
X(K) = X(K) - CONJG(AS(J))*ACC
680 CONTINUE
700 CONTINUE
ELSE IF (UNI) THEN
DO 740 I = 1, N
ACC = X(I)
DO 720 J = IA(I), IA(I+1) - 1
K = JA(J)
X(K) = X(K) - CONJG(AS(J))*ACC
720 CONTINUE
740 CONTINUE
END IF
END IF
END IF
RETURN
END