zhegst.f.html

来自「famous linear algebra library (LAPACK) p」· HTML 代码 · 共 284 行 · 第 1/2 页

HTML
284
字号
</span>      NB = <a name="ILAENV.124"></a><a href="hfy-index.html#ILAENV">ILAENV</a>( 1, <span class="string">'<a name="ZHEGST.124"></a><a href="zhegst.f.html#ZHEGST.1">ZHEGST</a>'</span>, UPLO, N, -1, -1, -1 )
<span class="comment">*</span><span class="comment">
</span>      IF( NB.LE.1 .OR. NB.GE.N ) THEN
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">        Use unblocked code
</span><span class="comment">*</span><span class="comment">
</span>         CALL <a name="ZHEGS2.130"></a><a href="zhegs2.f.html#ZHEGS2.1">ZHEGS2</a>( ITYPE, UPLO, N, A, LDA, B, LDB, INFO )
      ELSE
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">        Use blocked code
</span><span class="comment">*</span><span class="comment">
</span>         IF( ITYPE.EQ.1 ) THEN
            IF( UPPER ) THEN
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">              Compute inv(U')*A*inv(U)
</span><span class="comment">*</span><span class="comment">
</span>               DO 10 K = 1, N, NB
                  KB = MIN( N-K+1, NB )
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">                 Update the upper triangle of A(k:n,k:n)
</span><span class="comment">*</span><span class="comment">
</span>                  CALL <a name="ZHEGS2.145"></a><a href="zhegs2.f.html#ZHEGS2.1">ZHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
     $                         B( K, K ), LDB, INFO )
                  IF( K+KB.LE.N ) THEN
                     CALL ZTRSM( <span class="string">'Left'</span>, UPLO, <span class="string">'Conjugate transpose'</span>,
     $                           <span class="string">'Non-unit'</span>, KB, N-K-KB+1, CONE,
     $                           B( K, K ), LDB, A( K, K+KB ), LDA )
                     CALL ZHEMM( <span class="string">'Left'</span>, UPLO, KB, N-K-KB+1, -HALF,
     $                           A( K, K ), LDA, B( K, K+KB ), LDB,
     $                           CONE, A( K, K+KB ), LDA )
                     CALL ZHER2K( UPLO, <span class="string">'Conjugate transpose'</span>, N-K-KB+1,
     $                            KB, -CONE, A( K, K+KB ), LDA,
     $                            B( K, K+KB ), LDB, ONE,
     $                            A( K+KB, K+KB ), LDA )
                     CALL ZHEMM( <span class="string">'Left'</span>, UPLO, KB, N-K-KB+1, -HALF,
     $                           A( K, K ), LDA, B( K, K+KB ), LDB,
     $                           CONE, A( K, K+KB ), LDA )
                     CALL ZTRSM( <span class="string">'Right'</span>, UPLO, <span class="string">'No transpose'</span>,
     $                           <span class="string">'Non-unit'</span>, KB, N-K-KB+1, CONE,
     $                           B( K+KB, K+KB ), LDB, A( K, K+KB ),
     $                           LDA )
                  END IF
   10          CONTINUE
            ELSE
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">              Compute inv(L)*A*inv(L')
</span><span class="comment">*</span><span class="comment">
</span>               DO 20 K = 1, N, NB
                  KB = MIN( N-K+1, NB )
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">                 Update the lower triangle of A(k:n,k:n)
</span><span class="comment">*</span><span class="comment">
</span>                  CALL <a name="ZHEGS2.176"></a><a href="zhegs2.f.html#ZHEGS2.1">ZHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
     $                         B( K, K ), LDB, INFO )
                  IF( K+KB.LE.N ) THEN
                     CALL ZTRSM( <span class="string">'Right'</span>, UPLO, <span class="string">'Conjugate transpose'</span>,
     $                           <span class="string">'Non-unit'</span>, N-K-KB+1, KB, CONE,
     $                           B( K, K ), LDB, A( K+KB, K ), LDA )
                     CALL ZHEMM( <span class="string">'Right'</span>, UPLO, N-K-KB+1, KB, -HALF,
     $                           A( K, K ), LDA, B( K+KB, K ), LDB,
     $                           CONE, A( K+KB, K ), LDA )
                     CALL ZHER2K( UPLO, <span class="string">'No transpose'</span>, N-K-KB+1, KB,
     $                            -CONE, A( K+KB, K ), LDA,
     $                            B( K+KB, K ), LDB, ONE,
     $                            A( K+KB, K+KB ), LDA )
                     CALL ZHEMM( <span class="string">'Right'</span>, UPLO, N-K-KB+1, KB, -HALF,
     $                           A( K, K ), LDA, B( K+KB, K ), LDB,
     $                           CONE, A( K+KB, K ), LDA )
                     CALL ZTRSM( <span class="string">'Left'</span>, UPLO, <span class="string">'No transpose'</span>,
     $                           <span class="string">'Non-unit'</span>, N-K-KB+1, KB, CONE,
     $                           B( K+KB, K+KB ), LDB, A( K+KB, K ),
     $                           LDA )
                  END IF
   20          CONTINUE
            END IF
         ELSE
            IF( UPPER ) THEN
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">              Compute U*A*U'
</span><span class="comment">*</span><span class="comment">
</span>               DO 30 K = 1, N, NB
                  KB = MIN( N-K+1, NB )
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">                 Update the upper triangle of A(1:k+kb-1,1:k+kb-1)
</span><span class="comment">*</span><span class="comment">
</span>                  CALL ZTRMM( <span class="string">'Left'</span>, UPLO, <span class="string">'No transpose'</span>, <span class="string">'Non-unit'</span>,
     $                        K-1, KB, CONE, B, LDB, A( 1, K ), LDA )
                  CALL ZHEMM( <span class="string">'Right'</span>, UPLO, K-1, KB, HALF, A( K, K ),
     $                        LDA, B( 1, K ), LDB, CONE, A( 1, K ),
     $                        LDA )
                  CALL ZHER2K( UPLO, <span class="string">'No transpose'</span>, K-1, KB, CONE,
     $                         A( 1, K ), LDA, B( 1, K ), LDB, ONE, A,
     $                         LDA )
                  CALL ZHEMM( <span class="string">'Right'</span>, UPLO, K-1, KB, HALF, A( K, K ),
     $                        LDA, B( 1, K ), LDB, CONE, A( 1, K ),
     $                        LDA )
                  CALL ZTRMM( <span class="string">'Right'</span>, UPLO, <span class="string">'Conjugate transpose'</span>,
     $                        <span class="string">'Non-unit'</span>, K-1, KB, CONE, B( K, K ), LDB,
     $                        A( 1, K ), LDA )
                  CALL <a name="ZHEGS2.223"></a><a href="zhegs2.f.html#ZHEGS2.1">ZHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
     $                         B( K, K ), LDB, INFO )
   30          CONTINUE
            ELSE
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">              Compute L'*A*L
</span><span class="comment">*</span><span class="comment">
</span>               DO 40 K = 1, N, NB
                  KB = MIN( N-K+1, NB )
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">                 Update the lower triangle of A(1:k+kb-1,1:k+kb-1)
</span><span class="comment">*</span><span class="comment">
</span>                  CALL ZTRMM( <span class="string">'Right'</span>, UPLO, <span class="string">'No transpose'</span>, <span class="string">'Non-unit'</span>,
     $                        KB, K-1, CONE, B, LDB, A( K, 1 ), LDA )
                  CALL ZHEMM( <span class="string">'Left'</span>, UPLO, KB, K-1, HALF, A( K, K ),
     $                        LDA, B( K, 1 ), LDB, CONE, A( K, 1 ),
     $                        LDA )
                  CALL ZHER2K( UPLO, <span class="string">'Conjugate transpose'</span>, K-1, KB,
     $                         CONE, A( K, 1 ), LDA, B( K, 1 ), LDB,
     $                         ONE, A, LDA )
                  CALL ZHEMM( <span class="string">'Left'</span>, UPLO, KB, K-1, HALF, A( K, K ),
     $                        LDA, B( K, 1 ), LDB, CONE, A( K, 1 ),
     $                        LDA )
                  CALL ZTRMM( <span class="string">'Left'</span>, UPLO, <span class="string">'Conjugate transpose'</span>,
     $                        <span class="string">'Non-unit'</span>, KB, K-1, CONE, B( K, K ), LDB,
     $                        A( K, 1 ), LDA )
                  CALL <a name="ZHEGS2.249"></a><a href="zhegs2.f.html#ZHEGS2.1">ZHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
     $                         B( K, K ), LDB, INFO )
   40          CONTINUE
            END IF
         END IF
      END IF
      RETURN
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     End of <a name="ZHEGST.257"></a><a href="zhegst.f.html#ZHEGST.1">ZHEGST</a>
</span><span class="comment">*</span><span class="comment">
</span>      END

</pre>

 </body>
</html>

⌨️ 快捷键说明

复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?