Annotation of rpl/lapack/lapack/zcposv.f, revision 1.10

1.7       bertrand    1: *> \brief <b> ZCPOSV computes the solution to system of linear equations A * X = B for PO matrices</b>
                      2: *
                      3: *  =========== DOCUMENTATION ===========
                      4: *
                      5: * Online html documentation available at 
                      6: *            http://www.netlib.org/lapack/explore-html/ 
                      7: *
                      8: *> \htmlonly
                      9: *> Download ZCPOSV + dependencies 
                     10: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/zcposv.f"> 
                     11: *> [TGZ]</a> 
                     12: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/zcposv.f"> 
                     13: *> [ZIP]</a> 
                     14: *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/zcposv.f"> 
                     15: *> [TXT]</a>
                     16: *> \endhtmlonly 
                     17: *
                     18: *  Definition:
                     19: *  ===========
                     20: *
                     21: *       SUBROUTINE ZCPOSV( UPLO, N, NRHS, A, LDA, B, LDB, X, LDX, WORK,
                     22: *                          SWORK, RWORK, ITER, INFO )
                     23: * 
                     24: *       .. Scalar Arguments ..
                     25: *       CHARACTER          UPLO
                     26: *       INTEGER            INFO, ITER, LDA, LDB, LDX, N, NRHS
                     27: *       ..
                     28: *       .. Array Arguments ..
                     29: *       DOUBLE PRECISION   RWORK( * )
                     30: *       COMPLEX            SWORK( * )
                     31: *       COMPLEX*16         A( LDA, * ), B( LDB, * ), WORK( N, * ),
                     32: *      $                   X( LDX, * )
                     33: *       ..
                     34: *  
                     35: *
                     36: *> \par Purpose:
                     37: *  =============
                     38: *>
                     39: *> \verbatim
                     40: *>
                     41: *> ZCPOSV computes the solution to a complex system of linear equations
                     42: *>    A * X = B,
                     43: *> where A is an N-by-N Hermitian positive definite matrix and X and B
                     44: *> are N-by-NRHS matrices.
                     45: *>
                     46: *> ZCPOSV first attempts to factorize the matrix in COMPLEX and use this
                     47: *> factorization within an iterative refinement procedure to produce a
                     48: *> solution with COMPLEX*16 normwise backward error quality (see below).
                     49: *> If the approach fails the method switches to a COMPLEX*16
                     50: *> factorization and solve.
                     51: *>
                     52: *> The iterative refinement is not going to be a winning strategy if
                     53: *> the ratio COMPLEX performance over COMPLEX*16 performance is too
                     54: *> small. A reasonable strategy should take the number of right-hand
                     55: *> sides and the size of the matrix into account. This might be done
                     56: *> with a call to ILAENV in the future. Up to now, we always try
                     57: *> iterative refinement.
                     58: *>
                     59: *> The iterative refinement process is stopped if
                     60: *>     ITER > ITERMAX
                     61: *> or for all the RHS we have:
                     62: *>     RNRM < SQRT(N)*XNRM*ANRM*EPS*BWDMAX
                     63: *> where
                     64: *>     o ITER is the number of the current iteration in the iterative
                     65: *>       refinement process
                     66: *>     o RNRM is the infinity-norm of the residual
                     67: *>     o XNRM is the infinity-norm of the solution
                     68: *>     o ANRM is the infinity-operator-norm of the matrix A
                     69: *>     o EPS is the machine epsilon returned by DLAMCH('Epsilon')
                     70: *> The value ITERMAX and BWDMAX are fixed to 30 and 1.0D+00
                     71: *> respectively.
                     72: *> \endverbatim
                     73: *
                     74: *  Arguments:
                     75: *  ==========
                     76: *
                     77: *> \param[in] UPLO
                     78: *> \verbatim
                     79: *>          UPLO is CHARACTER*1
                     80: *>          = 'U':  Upper triangle of A is stored;
                     81: *>          = 'L':  Lower triangle of A is stored.
                     82: *> \endverbatim
                     83: *>
                     84: *> \param[in] N
                     85: *> \verbatim
                     86: *>          N is INTEGER
                     87: *>          The number of linear equations, i.e., the order of the
                     88: *>          matrix A.  N >= 0.
                     89: *> \endverbatim
                     90: *>
                     91: *> \param[in] NRHS
                     92: *> \verbatim
                     93: *>          NRHS is INTEGER
                     94: *>          The number of right hand sides, i.e., the number of columns
                     95: *>          of the matrix B.  NRHS >= 0.
                     96: *> \endverbatim
                     97: *>
                     98: *> \param[in,out] A
                     99: *> \verbatim
                    100: *>          A is COMPLEX*16 array,
                    101: *>          dimension (LDA,N)
                    102: *>          On entry, the Hermitian matrix A. If UPLO = 'U', the leading
                    103: *>          N-by-N upper triangular part of A contains the upper
                    104: *>          triangular part of the matrix A, and the strictly lower
                    105: *>          triangular part of A is not referenced.  If UPLO = 'L', the
                    106: *>          leading N-by-N lower triangular part of A contains the lower
                    107: *>          triangular part of the matrix A, and the strictly upper
                    108: *>          triangular part of A is not referenced.
                    109: *>
                    110: *>          Note that the imaginary parts of the diagonal
                    111: *>          elements need not be set and are assumed to be zero.
                    112: *>
                    113: *>          On exit, if iterative refinement has been successfully used
                    114: *>          (INFO.EQ.0 and ITER.GE.0, see description below), then A is
                    115: *>          unchanged, if double precision factorization has been used
                    116: *>          (INFO.EQ.0 and ITER.LT.0, see description below), then the
                    117: *>          array A contains the factor U or L from the Cholesky
                    118: *>          factorization A = U**H*U or A = L*L**H.
                    119: *> \endverbatim
                    120: *>
                    121: *> \param[in] LDA
                    122: *> \verbatim
                    123: *>          LDA is INTEGER
                    124: *>          The leading dimension of the array A.  LDA >= max(1,N).
                    125: *> \endverbatim
                    126: *>
                    127: *> \param[in] B
                    128: *> \verbatim
                    129: *>          B is COMPLEX*16 array, dimension (LDB,NRHS)
                    130: *>          The N-by-NRHS right hand side matrix B.
                    131: *> \endverbatim
                    132: *>
                    133: *> \param[in] LDB
                    134: *> \verbatim
                    135: *>          LDB is INTEGER
                    136: *>          The leading dimension of the array B.  LDB >= max(1,N).
                    137: *> \endverbatim
                    138: *>
                    139: *> \param[out] X
                    140: *> \verbatim
                    141: *>          X is COMPLEX*16 array, dimension (LDX,NRHS)
                    142: *>          If INFO = 0, the N-by-NRHS solution matrix X.
                    143: *> \endverbatim
                    144: *>
                    145: *> \param[in] LDX
                    146: *> \verbatim
                    147: *>          LDX is INTEGER
                    148: *>          The leading dimension of the array X.  LDX >= max(1,N).
                    149: *> \endverbatim
                    150: *>
                    151: *> \param[out] WORK
                    152: *> \verbatim
                    153: *>          WORK is COMPLEX*16 array, dimension (N*NRHS)
                    154: *>          This array is used to hold the residual vectors.
                    155: *> \endverbatim
                    156: *>
                    157: *> \param[out] SWORK
                    158: *> \verbatim
                    159: *>          SWORK is COMPLEX array, dimension (N*(N+NRHS))
                    160: *>          This array is used to use the single precision matrix and the
                    161: *>          right-hand sides or solutions in single precision.
                    162: *> \endverbatim
                    163: *>
                    164: *> \param[out] RWORK
                    165: *> \verbatim
                    166: *>          RWORK is DOUBLE PRECISION array, dimension (N)
                    167: *> \endverbatim
                    168: *>
                    169: *> \param[out] ITER
                    170: *> \verbatim
                    171: *>          ITER is INTEGER
                    172: *>          < 0: iterative refinement has failed, COMPLEX*16
                    173: *>               factorization has been performed
                    174: *>               -1 : the routine fell back to full precision for
                    175: *>                    implementation- or machine-specific reasons
                    176: *>               -2 : narrowing the precision induced an overflow,
                    177: *>                    the routine fell back to full precision
                    178: *>               -3 : failure of CPOTRF
                    179: *>               -31: stop the iterative refinement after the 30th
                    180: *>                    iterations
                    181: *>          > 0: iterative refinement has been sucessfully used.
                    182: *>               Returns the number of iterations
                    183: *> \endverbatim
                    184: *>
                    185: *> \param[out] INFO
                    186: *> \verbatim
                    187: *>          INFO is INTEGER
                    188: *>          = 0:  successful exit
                    189: *>          < 0:  if INFO = -i, the i-th argument had an illegal value
                    190: *>          > 0:  if INFO = i, the leading minor of order i of
                    191: *>                (COMPLEX*16) A is not positive definite, so the
                    192: *>                factorization could not be completed, and the solution
                    193: *>                has not been computed.
                    194: *> \endverbatim
                    195: *
                    196: *  Authors:
                    197: *  ========
                    198: *
                    199: *> \author Univ. of Tennessee 
                    200: *> \author Univ. of California Berkeley 
                    201: *> \author Univ. of Colorado Denver 
                    202: *> \author NAG Ltd. 
                    203: *
                    204: *> \date November 2011
                    205: *
                    206: *> \ingroup complex16POsolve
                    207: *
                    208: *  =====================================================================
1.1       bertrand  209:       SUBROUTINE ZCPOSV( UPLO, N, NRHS, A, LDA, B, LDB, X, LDX, WORK,
1.6       bertrand  210:      $                   SWORK, RWORK, ITER, INFO )
1.1       bertrand  211: *
1.7       bertrand  212: *  -- LAPACK driver routine (version 3.4.0) --
1.1       bertrand  213: *  -- LAPACK is a software package provided by Univ. of Tennessee,    --
                    214: *  -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
1.7       bertrand  215: *     November 2011
                    216: *
1.1       bertrand  217: *     .. Scalar Arguments ..
                    218:       CHARACTER          UPLO
                    219:       INTEGER            INFO, ITER, LDA, LDB, LDX, N, NRHS
                    220: *     ..
                    221: *     .. Array Arguments ..
                    222:       DOUBLE PRECISION   RWORK( * )
                    223:       COMPLEX            SWORK( * )
                    224:       COMPLEX*16         A( LDA, * ), B( LDB, * ), WORK( N, * ),
1.6       bertrand  225:      $                   X( LDX, * )
1.1       bertrand  226: *     ..
                    227: *
1.6       bertrand  228: *  =====================================================================
1.1       bertrand  229: *
                    230: *     .. Parameters ..
                    231:       LOGICAL            DOITREF
                    232:       PARAMETER          ( DOITREF = .TRUE. )
                    233: *
                    234:       INTEGER            ITERMAX
                    235:       PARAMETER          ( ITERMAX = 30 )
                    236: *
                    237:       DOUBLE PRECISION   BWDMAX
                    238:       PARAMETER          ( BWDMAX = 1.0E+00 )
                    239: *
                    240:       COMPLEX*16         NEGONE, ONE
                    241:       PARAMETER          ( NEGONE = ( -1.0D+00, 0.0D+00 ),
1.6       bertrand  242:      $                   ONE = ( 1.0D+00, 0.0D+00 ) )
1.1       bertrand  243: *
                    244: *     .. Local Scalars ..
                    245:       INTEGER            I, IITER, PTSA, PTSX
                    246:       DOUBLE PRECISION   ANRM, CTE, EPS, RNRM, XNRM
                    247:       COMPLEX*16         ZDUM
                    248: *
                    249: *     .. External Subroutines ..
                    250:       EXTERNAL           ZAXPY, ZHEMM, ZLACPY, ZLAT2C, ZLAG2C, CLAG2Z,
1.6       bertrand  251:      $                   CPOTRF, CPOTRS, XERBLA
1.1       bertrand  252: *     ..
                    253: *     .. External Functions ..
                    254:       INTEGER            IZAMAX
                    255:       DOUBLE PRECISION   DLAMCH, ZLANHE
                    256:       LOGICAL            LSAME
                    257:       EXTERNAL           IZAMAX, DLAMCH, ZLANHE, LSAME
                    258: *     ..
                    259: *     .. Intrinsic Functions ..
                    260:       INTRINSIC          ABS, DBLE, MAX, SQRT
                    261: *     .. Statement Functions ..
                    262:       DOUBLE PRECISION   CABS1
                    263: *     ..
                    264: *     .. Statement Function definitions ..
                    265:       CABS1( ZDUM ) = ABS( DBLE( ZDUM ) ) + ABS( DIMAG( ZDUM ) )
                    266: *     ..
                    267: *     .. Executable Statements ..
                    268: *
                    269:       INFO = 0
                    270:       ITER = 0
                    271: *
                    272: *     Test the input parameters.
                    273: *
                    274:       IF( .NOT.LSAME( UPLO, 'U' ) .AND. .NOT.LSAME( UPLO, 'L' ) ) THEN
                    275:          INFO = -1
                    276:       ELSE IF( N.LT.0 ) THEN
                    277:          INFO = -2
                    278:       ELSE IF( NRHS.LT.0 ) THEN
                    279:          INFO = -3
                    280:       ELSE IF( LDA.LT.MAX( 1, N ) ) THEN
                    281:          INFO = -5
                    282:       ELSE IF( LDB.LT.MAX( 1, N ) ) THEN
                    283:          INFO = -7
                    284:       ELSE IF( LDX.LT.MAX( 1, N ) ) THEN
                    285:          INFO = -9
                    286:       END IF
                    287:       IF( INFO.NE.0 ) THEN
                    288:          CALL XERBLA( 'ZCPOSV', -INFO )
                    289:          RETURN
                    290:       END IF
                    291: *
                    292: *     Quick return if (N.EQ.0).
                    293: *
                    294:       IF( N.EQ.0 )
1.6       bertrand  295:      $   RETURN
1.1       bertrand  296: *
                    297: *     Skip single precision iterative refinement if a priori slower
                    298: *     than double precision factorization.
                    299: *
                    300:       IF( .NOT.DOITREF ) THEN
                    301:          ITER = -1
                    302:          GO TO 40
                    303:       END IF
                    304: *
                    305: *     Compute some constants.
                    306: *
                    307:       ANRM = ZLANHE( 'I', UPLO, N, A, LDA, RWORK )
                    308:       EPS = DLAMCH( 'Epsilon' )
                    309:       CTE = ANRM*EPS*SQRT( DBLE( N ) )*BWDMAX
                    310: *
                    311: *     Set the indices PTSA, PTSX for referencing SA and SX in SWORK.
                    312: *
                    313:       PTSA = 1
                    314:       PTSX = PTSA + N*N
                    315: *
                    316: *     Convert B from double precision to single precision and store the
                    317: *     result in SX.
                    318: *
                    319:       CALL ZLAG2C( N, NRHS, B, LDB, SWORK( PTSX ), N, INFO )
                    320: *
                    321:       IF( INFO.NE.0 ) THEN
                    322:          ITER = -2
                    323:          GO TO 40
                    324:       END IF
                    325: *
                    326: *     Convert A from double precision to single precision and store the
                    327: *     result in SA.
                    328: *
                    329:       CALL ZLAT2C( UPLO, N, A, LDA, SWORK( PTSA ), N, INFO )
                    330: *
                    331:       IF( INFO.NE.0 ) THEN
                    332:          ITER = -2
                    333:          GO TO 40
                    334:       END IF
                    335: *
                    336: *     Compute the Cholesky factorization of SA.
                    337: *
                    338:       CALL CPOTRF( UPLO, N, SWORK( PTSA ), N, INFO )
                    339: *
                    340:       IF( INFO.NE.0 ) THEN
                    341:          ITER = -3
                    342:          GO TO 40
                    343:       END IF
                    344: *
                    345: *     Solve the system SA*SX = SB.
                    346: *
                    347:       CALL CPOTRS( UPLO, N, NRHS, SWORK( PTSA ), N, SWORK( PTSX ), N,
1.6       bertrand  348:      $             INFO )
1.1       bertrand  349: *
                    350: *     Convert SX back to COMPLEX*16
                    351: *
                    352:       CALL CLAG2Z( N, NRHS, SWORK( PTSX ), N, X, LDX, INFO )
                    353: *
                    354: *     Compute R = B - AX (R is WORK).
                    355: *
                    356:       CALL ZLACPY( 'All', N, NRHS, B, LDB, WORK, N )
                    357: *
                    358:       CALL ZHEMM( 'Left', UPLO, N, NRHS, NEGONE, A, LDA, X, LDX, ONE,
1.6       bertrand  359:      $            WORK, N )
1.1       bertrand  360: *
                    361: *     Check whether the NRHS normwise backward errors satisfy the
                    362: *     stopping criterion. If yes, set ITER=0 and return.
                    363: *
                    364:       DO I = 1, NRHS
                    365:          XNRM = CABS1( X( IZAMAX( N, X( 1, I ), 1 ), I ) )
                    366:          RNRM = CABS1( WORK( IZAMAX( N, WORK( 1, I ), 1 ), I ) )
                    367:          IF( RNRM.GT.XNRM*CTE )
1.6       bertrand  368:      $      GO TO 10
1.1       bertrand  369:       END DO
                    370: *
                    371: *     If we are here, the NRHS normwise backward errors satisfy the
                    372: *     stopping criterion. We are good to exit.
                    373: *
                    374:       ITER = 0
                    375:       RETURN
                    376: *
                    377:    10 CONTINUE
                    378: *
                    379:       DO 30 IITER = 1, ITERMAX
                    380: *
                    381: *        Convert R (in WORK) from double precision to single precision
                    382: *        and store the result in SX.
                    383: *
                    384:          CALL ZLAG2C( N, NRHS, WORK, N, SWORK( PTSX ), N, INFO )
                    385: *
                    386:          IF( INFO.NE.0 ) THEN
                    387:             ITER = -2
                    388:             GO TO 40
                    389:          END IF
                    390: *
                    391: *        Solve the system SA*SX = SR.
                    392: *
                    393:          CALL CPOTRS( UPLO, N, NRHS, SWORK( PTSA ), N, SWORK( PTSX ), N,
1.6       bertrand  394:      $                INFO )
1.1       bertrand  395: *
                    396: *        Convert SX back to double precision and update the current
                    397: *        iterate.
                    398: *
                    399:          CALL CLAG2Z( N, NRHS, SWORK( PTSX ), N, WORK, N, INFO )
                    400: *
                    401:          DO I = 1, NRHS
                    402:             CALL ZAXPY( N, ONE, WORK( 1, I ), 1, X( 1, I ), 1 )
                    403:          END DO
                    404: *
                    405: *        Compute R = B - AX (R is WORK).
                    406: *
                    407:          CALL ZLACPY( 'All', N, NRHS, B, LDB, WORK, N )
                    408: *
                    409:          CALL ZHEMM( 'L', UPLO, N, NRHS, NEGONE, A, LDA, X, LDX, ONE,
1.6       bertrand  410:      $               WORK, N )
1.1       bertrand  411: *
                    412: *        Check whether the NRHS normwise backward errors satisfy the
                    413: *        stopping criterion. If yes, set ITER=IITER>0 and return.
                    414: *
                    415:          DO I = 1, NRHS
                    416:             XNRM = CABS1( X( IZAMAX( N, X( 1, I ), 1 ), I ) )
                    417:             RNRM = CABS1( WORK( IZAMAX( N, WORK( 1, I ), 1 ), I ) )
                    418:             IF( RNRM.GT.XNRM*CTE )
1.6       bertrand  419:      $         GO TO 20
1.1       bertrand  420:          END DO
                    421: *
                    422: *        If we are here, the NRHS normwise backward errors satisfy the
                    423: *        stopping criterion, we are good to exit.
                    424: *
                    425:          ITER = IITER
                    426: *
                    427:          RETURN
                    428: *
                    429:    20    CONTINUE
                    430: *
                    431:    30 CONTINUE
                    432: *
                    433: *     If we are at this place of the code, this is because we have
                    434: *     performed ITER=ITERMAX iterations and never satisified the
                    435: *     stopping criterion, set up the ITER flag accordingly and follow
                    436: *     up on double precision routine.
                    437: *
                    438:       ITER = -ITERMAX - 1
                    439: *
                    440:    40 CONTINUE
                    441: *
                    442: *     Single-precision iterative refinement failed to converge to a
                    443: *     satisfactory solution, so we resort to double precision.
                    444: *
                    445:       CALL ZPOTRF( UPLO, N, A, LDA, INFO )
                    446: *
                    447:       IF( INFO.NE.0 )
1.6       bertrand  448:      $   RETURN
1.1       bertrand  449: *
                    450:       CALL ZLACPY( 'All', N, NRHS, B, LDB, X, LDX )
                    451:       CALL ZPOTRS( UPLO, N, NRHS, A, LDA, X, LDX, INFO )
                    452: *
                    453:       RETURN
                    454: *
                    455: *     End of ZCPOSV.
                    456: *
                    457:       END

CVSweb interface <joel.bertrand@systella.fr>