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.
145 lines
4.7 KiB
Fortran
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 SJADMM(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_spk_) ALPHA, BETA
|
|
INTEGER K, LDB, LDC, M, N,IERROR
|
|
CHARACTER TRANSA
|
|
C .. Array Arguments ..
|
|
real(psb_spk_) 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)
|
|
|
|
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 SJADMV(DIAG,M,N,ALPHA,IA(PNG),
|
|
+ AR,JA,IA(PJA),IA(PIA),B(1,1),BETA,C(1,1),IERROR)
|
|
ELSE
|
|
CALL SJADMV2(DIAG,M,N,ALPHA,IA(PNG),
|
|
+ AR,JA,IA(PJA),IA(PIA),B(1,1),LDB,BETA,C(1,1),LDC,IERROR)
|
|
CALL SJADMV3(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 SJADMV2(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 SJADMV3(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 SJADMV4(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
|