LAPACK 3.3.0

spftrf.f

Go to the documentation of this file.
00001       SUBROUTINE SPFTRF( TRANSR, UPLO, N, A, INFO )
00002 *
00003 *  -- LAPACK routine (version 3.3.0)                                    --
00004 *
00005 *  -- Contributed by Fred Gustavson of the IBM Watson Research Center --
00006 *     November 2010
00007 *
00008 *  -- LAPACK is a software package provided by Univ. of Tennessee,    --
00009 *  -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
00010 *
00011 *     ..
00012 *     .. Scalar Arguments ..
00013       CHARACTER          TRANSR, UPLO
00014       INTEGER            N, INFO
00015 *     ..
00016 *     .. Array Arguments ..
00017       REAL               A( 0: * )
00018 *
00019 *  Purpose
00020 *  =======
00021 *
00022 *  SPFTRF computes the Cholesky factorization of a real symmetric
00023 *  positive definite matrix A.
00024 *
00025 *  The factorization has the form
00026 *     A = U**T * U,  if UPLO = 'U', or
00027 *     A = L  * L**T,  if UPLO = 'L',
00028 *  where U is an upper triangular matrix and L is lower triangular.
00029 *
00030 *  This is the block version of the algorithm, calling Level 3 BLAS.
00031 *
00032 *  Arguments
00033 *  =========
00034 *
00035 *  TRANSR    (input) CHARACTER*1
00036 *          = 'N':  The Normal TRANSR of RFP A is stored;
00037 *          = 'T':  The Transpose TRANSR of RFP A is stored.
00038 *
00039 *  UPLO    (input) CHARACTER*1
00040 *          = 'U':  Upper triangle of RFP A is stored;
00041 *          = 'L':  Lower triangle of RFP A is stored.
00042 *
00043 *  N       (input) INTEGER
00044 *          The order of the matrix A.  N >= 0.
00045 *
00046 *  A       (input/output) REAL array, dimension ( N*(N+1)/2 );
00047 *          On entry, the symmetric matrix A in RFP format. RFP format is
00048 *          described by TRANSR, UPLO, and N as follows: If TRANSR = 'N'
00049 *          then RFP A is (0:N,0:k-1) when N is even; k=N/2. RFP A is
00050 *          (0:N-1,0:k) when N is odd; k=N/2. IF TRANSR = 'T' then RFP is
00051 *          the transpose of RFP A as defined when
00052 *          TRANSR = 'N'. The contents of RFP A are defined by UPLO as
00053 *          follows: If UPLO = 'U' the RFP A contains the NT elements of
00054 *          upper packed A. If UPLO = 'L' the RFP A contains the elements
00055 *          of lower packed A. The LDA of RFP A is (N+1)/2 when TRANSR =
00056 *          'T'. When TRANSR is 'N' the LDA is N+1 when N is even and N
00057 *          is odd. See the Note below for more details.
00058 *
00059 *          On exit, if INFO = 0, the factor U or L from the Cholesky
00060 *          factorization RFP A = U**T*U or RFP A = L*L**T.
00061 *
00062 *  INFO    (output) INTEGER
00063 *          = 0:  successful exit
00064 *          < 0:  if INFO = -i, the i-th argument had an illegal value
00065 *          > 0:  if INFO = i, the leading minor of order i is not
00066 *                positive definite, and the factorization could not be
00067 *                completed.
00068 *
00069 *  Further Details
00070 *  ===============
00071 *
00072 *  We first consider Rectangular Full Packed (RFP) Format when N is
00073 *  even. We give an example where N = 6.
00074 *
00075 *      AP is Upper             AP is Lower
00076 *
00077 *   00 01 02 03 04 05       00
00078 *      11 12 13 14 15       10 11
00079 *         22 23 24 25       20 21 22
00080 *            33 34 35       30 31 32 33
00081 *               44 45       40 41 42 43 44
00082 *                  55       50 51 52 53 54 55
00083 *
00084 *
00085 *  Let TRANSR = 'N'. RFP holds AP as follows:
00086 *  For UPLO = 'U' the upper trapezoid A(0:5,0:2) consists of the last
00087 *  three columns of AP upper. The lower triangle A(4:6,0:2) consists of
00088 *  the transpose of the first three columns of AP upper.
00089 *  For UPLO = 'L' the lower trapezoid A(1:6,0:2) consists of the first
00090 *  three columns of AP lower. The upper triangle A(0:2,0:2) consists of
00091 *  the transpose of the last three columns of AP lower.
00092 *  This covers the case N even and TRANSR = 'N'.
00093 *
00094 *         RFP A                   RFP A
00095 *
00096 *        03 04 05                33 43 53
00097 *        13 14 15                00 44 54
00098 *        23 24 25                10 11 55
00099 *        33 34 35                20 21 22
00100 *        00 44 45                30 31 32
00101 *        01 11 55                40 41 42
00102 *        02 12 22                50 51 52
00103 *
00104 *  Now let TRANSR = 'T'. RFP A in both UPLO cases is just the
00105 *  transpose of RFP A above. One therefore gets:
00106 *
00107 *
00108 *           RFP A                   RFP A
00109 *
00110 *     03 13 23 33 00 01 02    33 00 10 20 30 40 50
00111 *     04 14 24 34 44 11 12    43 44 11 21 31 41 51
00112 *     05 15 25 35 45 55 22    53 54 55 22 32 42 52
00113 *
00114 *
00115 *  We then consider Rectangular Full Packed (RFP) Format when N is
00116 *  odd. We give an example where N = 5.
00117 *
00118 *     AP is Upper                 AP is Lower
00119 *
00120 *   00 01 02 03 04              00
00121 *      11 12 13 14              10 11
00122 *         22 23 24              20 21 22
00123 *            33 34              30 31 32 33
00124 *               44              40 41 42 43 44
00125 *
00126 *
00127 *  Let TRANSR = 'N'. RFP holds AP as follows:
00128 *  For UPLO = 'U' the upper trapezoid A(0:4,0:2) consists of the last
00129 *  three columns of AP upper. The lower triangle A(3:4,0:1) consists of
00130 *  the transpose of the first two columns of AP upper.
00131 *  For UPLO = 'L' the lower trapezoid A(0:4,0:2) consists of the first
00132 *  three columns of AP lower. The upper triangle A(0:1,1:2) consists of
00133 *  the transpose of the last two columns of AP lower.
00134 *  This covers the case N odd and TRANSR = 'N'.
00135 *
00136 *         RFP A                   RFP A
00137 *
00138 *        02 03 04                00 33 43
00139 *        12 13 14                10 11 44
00140 *        22 23 24                20 21 22
00141 *        00 33 34                30 31 32
00142 *        01 11 44                40 41 42
00143 *
00144 *  Now let TRANSR = 'T'. RFP A in both UPLO cases is just the
00145 *  transpose of RFP A above. One therefore gets:
00146 *
00147 *           RFP A                   RFP A
00148 *
00149 *     02 12 22 00 01             00 10 20 30 40 50
00150 *     03 13 23 33 11             33 11 21 31 41 51
00151 *     04 14 24 34 44             43 44 22 32 42 52
00152 *
00153 *  =====================================================================
00154 *
00155 *     .. Parameters ..
00156       REAL               ONE
00157       PARAMETER          ( ONE = 1.0E+0 )
00158 *     ..
00159 *     .. Local Scalars ..
00160       LOGICAL            LOWER, NISODD, NORMALTRANSR
00161       INTEGER            N1, N2, K
00162 *     ..
00163 *     .. External Functions ..
00164       LOGICAL            LSAME
00165       EXTERNAL           LSAME
00166 *     ..
00167 *     .. External Subroutines ..
00168       EXTERNAL           XERBLA, SSYRK, SPOTRF, STRSM
00169 *     ..
00170 *     .. Intrinsic Functions ..
00171       INTRINSIC          MOD
00172 *     ..
00173 *     .. Executable Statements ..
00174 *
00175 *     Test the input parameters.
00176 *
00177       INFO = 0
00178       NORMALTRANSR = LSAME( TRANSR, 'N' )
00179       LOWER = LSAME( UPLO, 'L' )
00180       IF( .NOT.NORMALTRANSR .AND. .NOT.LSAME( TRANSR, 'T' ) ) THEN
00181          INFO = -1
00182       ELSE IF( .NOT.LOWER .AND. .NOT.LSAME( UPLO, 'U' ) ) THEN
00183          INFO = -2
00184       ELSE IF( N.LT.0 ) THEN
00185          INFO = -3
00186       END IF
00187       IF( INFO.NE.0 ) THEN
00188          CALL XERBLA( 'SPFTRF', -INFO )
00189          RETURN
00190       END IF
00191 *
00192 *     Quick return if possible
00193 *
00194       IF( N.EQ.0 )
00195      +   RETURN
00196 *
00197 *     If N is odd, set NISODD = .TRUE.
00198 *     If N is even, set K = N/2 and NISODD = .FALSE.
00199 *
00200       IF( MOD( N, 2 ).EQ.0 ) THEN
00201          K = N / 2
00202          NISODD = .FALSE.
00203       ELSE
00204          NISODD = .TRUE.
00205       END IF
00206 *
00207 *     Set N1 and N2 depending on LOWER
00208 *
00209       IF( LOWER ) THEN
00210          N2 = N / 2
00211          N1 = N - N2
00212       ELSE
00213          N1 = N / 2
00214          N2 = N - N1
00215       END IF
00216 *
00217 *     start execution: there are eight cases
00218 *
00219       IF( NISODD ) THEN
00220 *
00221 *        N is odd
00222 *
00223          IF( NORMALTRANSR ) THEN
00224 *
00225 *           N is odd and TRANSR = 'N'
00226 *
00227             IF( LOWER ) THEN
00228 *
00229 *             SRPA for LOWER, NORMAL and N is odd ( a(0:n-1,0:n1-1) )
00230 *             T1 -> a(0,0), T2 -> a(0,1), S -> a(n1,0)
00231 *             T1 -> a(0), T2 -> a(n), S -> a(n1)
00232 *
00233                CALL SPOTRF( 'L', N1, A( 0 ), N, INFO )
00234                IF( INFO.GT.0 )
00235      +            RETURN
00236                CALL STRSM( 'R', 'L', 'T', 'N', N2, N1, ONE, A( 0 ), N,
00237      +                     A( N1 ), N )
00238                CALL SSYRK( 'U', 'N', N2, N1, -ONE, A( N1 ), N, ONE,
00239      +                     A( N ), N )
00240                CALL SPOTRF( 'U', N2, A( N ), N, INFO )
00241                IF( INFO.GT.0 )
00242      +            INFO = INFO + N1
00243 *
00244             ELSE
00245 *
00246 *             SRPA for UPPER, NORMAL and N is odd ( a(0:n-1,0:n2-1)
00247 *             T1 -> a(n1+1,0), T2 -> a(n1,0), S -> a(0,0)
00248 *             T1 -> a(n2), T2 -> a(n1), S -> a(0)
00249 *
00250                CALL SPOTRF( 'L', N1, A( N2 ), N, INFO )
00251                IF( INFO.GT.0 )
00252      +            RETURN
00253                CALL STRSM( 'L', 'L', 'N', 'N', N1, N2, ONE, A( N2 ), N,
00254      +                     A( 0 ), N )
00255                CALL SSYRK( 'U', 'T', N2, N1, -ONE, A( 0 ), N, ONE,
00256      +                     A( N1 ), N )
00257                CALL SPOTRF( 'U', N2, A( N1 ), N, INFO )
00258                IF( INFO.GT.0 )
00259      +            INFO = INFO + N1
00260 *
00261             END IF
00262 *
00263          ELSE
00264 *
00265 *           N is odd and TRANSR = 'T'
00266 *
00267             IF( LOWER ) THEN
00268 *
00269 *              SRPA for LOWER, TRANSPOSE and N is odd
00270 *              T1 -> A(0,0) , T2 -> A(1,0) , S -> A(0,n1)
00271 *              T1 -> a(0+0) , T2 -> a(1+0) , S -> a(0+n1*n1); lda=n1
00272 *
00273                CALL SPOTRF( 'U', N1, A( 0 ), N1, INFO )
00274                IF( INFO.GT.0 )
00275      +            RETURN
00276                CALL STRSM( 'L', 'U', 'T', 'N', N1, N2, ONE, A( 0 ), N1,
00277      +                     A( N1*N1 ), N1 )
00278                CALL SSYRK( 'L', 'T', N2, N1, -ONE, A( N1*N1 ), N1, ONE,
00279      +                     A( 1 ), N1 )
00280                CALL SPOTRF( 'L', N2, A( 1 ), N1, INFO )
00281                IF( INFO.GT.0 )
00282      +            INFO = INFO + N1
00283 *
00284             ELSE
00285 *
00286 *              SRPA for UPPER, TRANSPOSE and N is odd
00287 *              T1 -> A(0,n1+1), T2 -> A(0,n1), S -> A(0,0)
00288 *              T1 -> a(n2*n2), T2 -> a(n1*n2), S -> a(0); lda = n2
00289 *
00290                CALL SPOTRF( 'U', N1, A( N2*N2 ), N2, INFO )
00291                IF( INFO.GT.0 )
00292      +            RETURN
00293                CALL STRSM( 'R', 'U', 'N', 'N', N2, N1, ONE, A( N2*N2 ),
00294      +                     N2, A( 0 ), N2 )
00295                CALL SSYRK( 'L', 'N', N2, N1, -ONE, A( 0 ), N2, ONE,
00296      +                     A( N1*N2 ), N2 )
00297                CALL SPOTRF( 'L', N2, A( N1*N2 ), N2, INFO )
00298                IF( INFO.GT.0 )
00299      +            INFO = INFO + N1
00300 *
00301             END IF
00302 *
00303          END IF
00304 *
00305       ELSE
00306 *
00307 *        N is even
00308 *
00309          IF( NORMALTRANSR ) THEN
00310 *
00311 *           N is even and TRANSR = 'N'
00312 *
00313             IF( LOWER ) THEN
00314 *
00315 *              SRPA for LOWER, NORMAL, and N is even ( a(0:n,0:k-1) )
00316 *              T1 -> a(1,0), T2 -> a(0,0), S -> a(k+1,0)
00317 *              T1 -> a(1), T2 -> a(0), S -> a(k+1)
00318 *
00319                CALL SPOTRF( 'L', K, A( 1 ), N+1, INFO )
00320                IF( INFO.GT.0 )
00321      +            RETURN
00322                CALL STRSM( 'R', 'L', 'T', 'N', K, K, ONE, A( 1 ), N+1,
00323      +                     A( K+1 ), N+1 )
00324                CALL SSYRK( 'U', 'N', K, K, -ONE, A( K+1 ), N+1, ONE,
00325      +                     A( 0 ), N+1 )
00326                CALL SPOTRF( 'U', K, A( 0 ), N+1, INFO )
00327                IF( INFO.GT.0 )
00328      +            INFO = INFO + K
00329 *
00330             ELSE
00331 *
00332 *              SRPA for UPPER, NORMAL, and N is even ( a(0:n,0:k-1) )
00333 *              T1 -> a(k+1,0) ,  T2 -> a(k,0),   S -> a(0,0)
00334 *              T1 -> a(k+1), T2 -> a(k), S -> a(0)
00335 *
00336                CALL SPOTRF( 'L', K, A( K+1 ), N+1, INFO )
00337                IF( INFO.GT.0 )
00338      +            RETURN
00339                CALL STRSM( 'L', 'L', 'N', 'N', K, K, ONE, A( K+1 ),
00340      +                     N+1, A( 0 ), N+1 )
00341                CALL SSYRK( 'U', 'T', K, K, -ONE, A( 0 ), N+1, ONE,
00342      +                     A( K ), N+1 )
00343                CALL SPOTRF( 'U', K, A( K ), N+1, INFO )
00344                IF( INFO.GT.0 )
00345      +            INFO = INFO + K
00346 *
00347             END IF
00348 *
00349          ELSE
00350 *
00351 *           N is even and TRANSR = 'T'
00352 *
00353             IF( LOWER ) THEN
00354 *
00355 *              SRPA for LOWER, TRANSPOSE and N is even (see paper)
00356 *              T1 -> B(0,1), T2 -> B(0,0), S -> B(0,k+1)
00357 *              T1 -> a(0+k), T2 -> a(0+0), S -> a(0+k*(k+1)); lda=k
00358 *
00359                CALL SPOTRF( 'U', K, A( 0+K ), K, INFO )
00360                IF( INFO.GT.0 )
00361      +            RETURN
00362                CALL STRSM( 'L', 'U', 'T', 'N', K, K, ONE, A( K ), N1,
00363      +                     A( K*( K+1 ) ), K )
00364                CALL SSYRK( 'L', 'T', K, K, -ONE, A( K*( K+1 ) ), K, ONE,
00365      +                     A( 0 ), K )
00366                CALL SPOTRF( 'L', K, A( 0 ), K, INFO )
00367                IF( INFO.GT.0 )
00368      +            INFO = INFO + K
00369 *
00370             ELSE
00371 *
00372 *              SRPA for UPPER, TRANSPOSE and N is even (see paper)
00373 *              T1 -> B(0,k+1),     T2 -> B(0,k),   S -> B(0,0)
00374 *              T1 -> a(0+k*(k+1)), T2 -> a(0+k*k), S -> a(0+0)); lda=k
00375 *
00376                CALL SPOTRF( 'U', K, A( K*( K+1 ) ), K, INFO )
00377                IF( INFO.GT.0 )
00378      +            RETURN
00379                CALL STRSM( 'R', 'U', 'N', 'N', K, K, ONE,
00380      +                     A( K*( K+1 ) ), K, A( 0 ), K )
00381                CALL SSYRK( 'L', 'N', K, K, -ONE, A( 0 ), K, ONE,
00382      +                     A( K*K ), K )
00383                CALL SPOTRF( 'L', K, A( K*K ), K, INFO )
00384                IF( INFO.GT.0 )
00385      +            INFO = INFO + K
00386 *
00387             END IF
00388 *
00389          END IF
00390 *
00391       END IF
00392 *
00393       RETURN
00394 *
00395 *     End of SPFTRF
00396 *
00397       END
 All Files Functions