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/jad/djadmm.f

145 lines
4.7 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
C
SUBROUTINE DJADMM(TRANSA,M,K,N,ALPHA,DESCRA,AR,
* JA,IA,B,LDB,BETA,C,LDC,WORK,IERROR)
use psb_const_mod
IMPLICIT NONE
C
C
C .. Scalar Arguments ..
real(psb_dpk_) ALPHA, BETA
INTEGER K, LDB, LDC, M, N,IERROR
CHARACTER TRANSA
C .. Array Arguments ..
real(psb_dpk_) AR(*), B(LDB,*), C(LDC,*), WORK(*)
INTEGER IA(*), JA(*)
CHARACTER DESCRA*11
C .. Local Scalars ..
integer PIA, PJA, PNG
INTEGER I, J, KB,NB, ERR_ACT
CHARACTER DIAG, TRANS
c .. Local Arrays ..
CHARACTER*20 NAME
INTEGER INT_VAL(5)
C .. Executable Statements ..
C
NAME = 'DJADMM\0'
IERROR = 0
CALL FCPSB_ERRACTIONSAVE(ERR_ACT)
IF (DESCRA(1:1).EQ.'G') TRANS = TRANSA
IF (DESCRA(1:1).EQ.'S' .AND. DESCRA(2:2).EQ.'U') TRANS = 'U'
IF (DESCRA(1:1).EQ.'S' .AND. DESCRA(2:2).EQ.'L') TRANS = 'L'
IF (DESCRA(1:1).EQ.'D') THEN
IF (DESCRA(3:3).EQ.'U') THEN
DO 40 I = 1, K
DO 20 J = 1, M
C(J,I) = BETA*C(J,I) + ALPHA*B(J,I)
20 CONTINUE
40 CONTINUE
ELSE
DO 80 I = 1, K
DO 60 J = 1, M
C(J,I) = BETA*C(J,I) + ALPHA*AR(J)*B(J,I)
60 CONTINUE
80 CONTINUE
END IF
RETURN
END IF
IF (TRANS.EQ.'T'.or.TRANS.EQ.'C') THEN
IERROR = 3015
RETURN
ENDIF
C
IF (DESCRA(3:3).EQ.'N') DIAG = 'N'
IF (DESCRA(3:3).EQ.'U') DIAG = 'U'
PNG = IA(1)
PIA = IA(2)
PJA = IA(3)
* write(6,*)'M N K',m,n,k
NB = 4
KB = MOD(K,NB)
SELECT CASE(KB)
CASE (1)
IF (K==1) THEN
CALL DJADMV(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,1),BETA,C(1,1),IERROR)
ELSE
CALL DJADMV2(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,1),LDB,BETA,C(1,1),LDC,IERROR)
CALL DJADMV3(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,3),LDB,BETA,C(1,3),LDC,IERROR)
KB = KB + 4
ENDIF
CASE(2)
CALL DJADMV2(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,1),LDB,BETA,C(1,1),LDC,IERROR)
CASE(3)
CALL DJADMV3(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,1),LDB,BETA,C(1,1),LDC,IERROR)
END SELECT
IF(IERROR.NE.0) THEN
INT_VAL(1)=IERROR
CALL FCPSB_ERRPUSH(4012,NAME,INT_VAL)
GOTO 9999
END IF
DO I=KB+1,K,NB
CALL DJADMV4(DIAG,M,N,ALPHA,IA(PNG),
+ AR,JA,IA(PJA),IA(PIA),B(1,I),LDB,BETA,C(1,I),LDC,IERROR)
END DO
IF(IERROR.NE.0) THEN
INT_VAL(1)=IERROR
CALL FCPSB_ERRPUSH(4012,NAME,INT_VAL)
GOTO 9999
END IF
CALL FCPSB_ERRACTIONRESTORE(ERR_ACT)
RETURN
9999 CONTINUE
CALL FCPSB_ERRACTIONRESTORE(ERR_ACT)
IF ( ERR_ACT .NE. 0 ) THEN
CALL FCPSB_SERROR()
RETURN
ENDIF
RETURN
END