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

238 lines
8.2 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 SUBROUTINE ZCSRMM(TRANSA,M,K,N,ALPHA,DESCRA,AR,
C * JA,IA,B,LDB,BETA,C,LDC,WORK,LWORK)
C
C Purpose
C =======
C
C Computing C <-- ALPHA A B + BETA C or
C C <-- ALPHA At B + BETA C or
C C <-- ALPHA Atc B + BETA C
C Called by ZSWMM
C This routine calls kernel for CSR data structure.
C
C Parameters
C ==========
C
C TRANSA - CHARACTER*1
C On entry TRANS specifies if the routine operates with matrix A
C or with the transpose of A as follows:
C TRANS = 'N' -> use matrix A
C TRANS = 'T' -> use A' (transpose of matrix A)
C TRANS = 'C' -> use conjugate transpose of A
C Unchanged on exit.
C
C N.B.: M, K for C matrix
C M, N for A matrix
C N, K for B matrix
C In the calling subroutine, ZSWMM, it was:
C M, N for C matrix
C M, K for A matrix
C K, N for B matrix
C Check the parameters order!
C
C M - INTEGER
C On entry: number of rows of matrix A (A') and
C number of rows of matrix C
C Unchanged on exit.
C
C K - INTEGER
C On entry: number of columns of matrix B
C and number of columns of matrix C.
C Unchanged on exit.
C
C N - INTEGER
C On entry: number of columns of matrix A (A') and
C number of rows of matrix B
C Unchanged on exit.
C
C ALPHA - COMPLEX*16
C On entry: multiplicative constant.
C Unchanged on exit.
C
C DESCRA - CHARACTER*1 array of DIMENSION (9)
C On entry DESCRA describes the characteristics of the input
C sparse matrix.
C Unchanged on exit.
C
C AR - COMPLEX*16 array of DIMENSION (*)
C On entry AR specifies the values of the input sparse
C matrix.
C Unchanged on exit.
C
C JA - INTEGER array of dimension (*)
C On entry JA holds integer information on input sparse
C matrix. Actual information will depend on data format used.
C Unchanged on exit.
C
C IA - INTEGER array of dimension (*)
C On entry IA holds integer information on input sparse
C matrix. Actual information will depend on data format used.
C Unchanged on exit.
C
C INFOA - INTEGER array of length 10.
C On entry can hold auxiliary information on input matrices
C formats or environment of subsequent calls.
C Might be changed on exit.
C
C B - COMPLEX*16 matrix of dimension (LDB,*)
C On entry: dense matrix.
C Unchanged on exit.
C
C LDB - INTEGER
C On entry: leading dimension of B
C Unchanged on exit.
C
C BETA - COMPLEX*16
C On entry: multiplicative constant.
C Unchanged on exit.
C
C C - COMPLEX*16 matrix of dimension (LDC,*)
C On entry: dense matrix.
C On exit is updated with the matrix-matrix product.
C
C LDC - INTEGER
C On entry: leading dimension of C
C Unchanged on exit.
C
C WORK - COMPLEX*16 array of dimension (LWORK)
C On entry: work area.
C On exit INT(WORK(1)) contains the minimum value
C for LWORK satisfying ZSWMM memory requirements.
C
C LWORK - INTEGER
C On entry LWORK specifies the dimension of WORK
C Unchanged on exit.
C
SUBROUTINE ZCSRMM(TRANSA,M,K,N,ALPHA,DESCRA,AR,
* JA,IA,B,LDB,BETA,C,LDC,WORK,LWORK)
use psb_const_mod
use psb_string_mod
C .. Scalar Arguments ..
complex(psb_dpk_) ALPHA, BETA
INTEGER K, LDB, LDC, M, N, LWORK
CHARACTER TRANSA
C .. Array Arguments ..
complex(psb_dpk_) AR(*), B(LDB,*), C(LDC,*), WORK(*)
INTEGER IA(*), JA(*)
CHARACTER DESCRA*11
C .. Local Scalars ..
INTEGER I, J
CHARACTER DIAG, TRANS
C .. External Subroutines ..
EXTERNAL ZSRMV
C .. Executable Statements ..
C
C
C IF (DESCRA(1).EQ.'G') TRANS = TRANSA
C
C .. Why to loose TRANSA for H, T, A matrices?
C
TRANS = TRANSA
C
IF ((psb_toupper(DESCRA(1:1)).EQ.'S').AND.
+ (psb_toupper(DESCRA(2:2)).EQ.'U')) THEN
IF (psb_toupper(TRANSA).EQ.'C') THEN
TRANS = 'V'
ELSE
TRANS = 'U'
ENDIF
ENDIF
IF ((psb_toupper(DESCRA(1:1)).EQ.'S').AND.
+ (psb_toupper(DESCRA(2:2)).EQ.'L')) THEN
IF (psb_toupper(TRANSA).EQ.'C') THEN
TRANS = 'M'
ELSE
TRANS = 'L'
ENDIF
ENDIF
C .. Diagonal matrix
IF (psb_toupper(DESCRA(1:1)).EQ.'D') THEN
C .. Diagonal matrix with unitary values
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
RETURN
C .. Diagonal matrix to be conjugated
ELSE IF (psb_toupper(TRANSA).EQ.'C') THEN
DO 80 I = 1, K
DO 60 J = 1, M
C(J,I) = BETA*C(J,I) + ALPHA *
+ CONJG(AR(J)) * B(J,I)
60 CONTINUE
80 CONTINUE
RETURN
C .. Generic diagonal matrix
ELSE
DO 91 I = 1, K
DO 90 J = 1, M
C(J,I) = BETA*C(J,I) + ALPHA *
+ AR(J) * B(J,I)
90 CONTINUE
91 CONTINUE
RETURN
ENDIF
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 or C=conjug(A')*B
C
C TRANS =
C 'N': Compute using A as it is.
C 'T': Compute using the transpose of A.
C 'C': Compute using conjugate transpose of A.
C 'U': A is symmetric, stored upper, compute as it is;
C (transposition makes no sense)
C 'V': A is symmetric, stored upper, to be conjugated.
C 'L': A is symmetric, stored lower, compute as it is.
C 'M': A is symmetric, stored lower, to be conjugated.
C DIAG =
C 'U': Unitary diagonal.
C 'N': Generic diagonal.
C
DO 100 I = 1, K
CALL ZSRMV(TRANS,DIAG,M,N,ALPHA,AR,JA,IA,B(1,I),
+ BETA,C(1,I),WORK)
100 CONTINUE
RETURN
END