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/dcsrmm.f

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