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.
148 lines
4.8 KiB
Fortran
148 lines
4.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
|
|
C
|
|
c
|
|
c What if a wrong DESCRA is passed?
|
|
c
|
|
c
|
|
*
|
|
*
|
|
SUBROUTINE DCSRMM(TRANSA,M,K,N,ALPHA,DESCRA,AR,
|
|
* JA,IA,B,LDB,BETA,C,LDC,WORK,LWORK,IERROR)
|
|
use psb_const_mod
|
|
use psb_string_mod
|
|
C .. Scalar Arguments ..
|
|
real(psb_dpk_) ALPHA, BETA
|
|
INTEGER K, LDB, LDC, M, N, LWORK,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 I, J, K4
|
|
CHARACTER DIAG, TRANS
|
|
c .. Local Arrays ..
|
|
CHARACTER*20 NAME
|
|
INTEGER INT_VAL(5)
|
|
|
|
C .. External Subroutines ..
|
|
EXTERNAL DCSRMV
|
|
C .. Executable Statements ..
|
|
C
|
|
C
|
|
NAME = 'DCSRMM\0'
|
|
IERROR = 0
|
|
CALL FCPSB_ERRACTIONSAVE(ERR_ACT)
|
|
C
|
|
IF (psb_toupper(DESCRA(1:1)).EQ.'G') TRANS = TRANSA
|
|
IF (psb_toupper(DESCRA(1:1)).EQ.'S' .AND.
|
|
+ psb_toupper(DESCRA(2:2)).EQ.'U') TRANS = 'U'
|
|
IF (psb_toupper(DESCRA(1:1)).EQ.'S' .AND.
|
|
+ psb_toupper(DESCRA(2:2)).EQ.'L') TRANS = 'L'
|
|
c
|
|
c Does DSRMV manage this case too?
|
|
c
|
|
IF (psb_toupper(DESCRA(1:1)).EQ.'D') THEN
|
|
IF (psb_toupper(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
|
|
C
|
|
IF (psb_toupper(DESCRA(3:3)).EQ.'N') DIAG = 'N'
|
|
IF (psb_toupper(DESCRA(3:3)).EQ.'U') DIAG = 'U'
|
|
C
|
|
C C = A*B OR C=A'*B
|
|
C
|
|
C
|
|
NB = 4
|
|
K4 = MOD(K,NB)
|
|
SELECT CASE(K4)
|
|
CASE(1)
|
|
IF (K==1) THEN
|
|
CALL DCSRMV(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,B(1,1),
|
|
+ BETA,C(1,1),WORK,LWORK,IERROR)
|
|
ELSE
|
|
CALL DCSRMV2(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,
|
|
+ B(1,1),LDB,BETA,C(1,1),LDC,WORK,LWORK,IERROR)
|
|
CALL DCSRMV3(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,
|
|
+ B(1,3),LDB,BETA,C(1,3),LDC,WORK,LWORK,IERROR)
|
|
K4 = K4 + 4
|
|
ENDIF
|
|
CASE(2)
|
|
CALL DCSRMV2(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,
|
|
+ B(1,1),LDB,BETA,C(1,1),LDC,WORK,LWORK,IERROR)
|
|
CASE(3)
|
|
CALL DCSRMV3(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,
|
|
+ B(1,1),LDB,BETA,C(1,1),LDC,WORK,LWORK,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=K4+1,K,NB
|
|
CALL DCSRMV4(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,
|
|
+ B(1,I),LDB,BETA,C(1,I),LDC,WORK,LWORK,IERROR)
|
|
ENDDO
|
|
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
|