chegst.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="CHEGST.124"></a><a href="chegst.f.html#CHEGST.1">CHEGST</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="CHEGS2.130"></a><a href="chegs2.f.html#CHEGS2.1">CHEGS2</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="CHEGS2.145"></a><a href="chegs2.f.html#CHEGS2.1">CHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
$ B( K, K ), LDB, INFO )
IF( K+KB.LE.N ) THEN
CALL CTRSM( <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 CHEMM( <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 CHER2K( 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 CHEMM( <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 CTRSM( <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="CHEGS2.176"></a><a href="chegs2.f.html#CHEGS2.1">CHEGS2</a>( ITYPE, UPLO, KB, A( K, K ), LDA,
$ B( K, K ), LDB, INFO )
IF( K+KB.LE.N ) THEN
CALL CTRSM( <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 CHEMM( <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 CHER2K( 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 CHEMM( <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 CTRSM( <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 CTRMM( <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 CHEMM( <span class="string">'Right'</span>, UPLO, K-1, KB, HALF, A( K, K ),
$ LDA, B( 1, K ), LDB, CONE, A( 1, K ),
$ LDA )
CALL CHER2K( UPLO, <span class="string">'No transpose'</span>, K-1, KB, CONE,
$ A( 1, K ), LDA, B( 1, K ), LDB, ONE, A,
$ LDA )
CALL CHEMM( <span class="string">'Right'</span>, UPLO, K-1, KB, HALF, A( K, K ),
$ LDA, B( 1, K ), LDB, CONE, A( 1, K ),
$ LDA )
CALL CTRMM( <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="CHEGS2.223"></a><a href="chegs2.f.html#CHEGS2.1">CHEGS2</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 CTRMM( <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 CHEMM( <span class="string">'Left'</span>, UPLO, KB, K-1, HALF, A( K, K ),
$ LDA, B( K, 1 ), LDB, CONE, A( K, 1 ),
$ LDA )
CALL CHER2K( UPLO, <span class="string">'Conjugate transpose'</span>, K-1, KB,
$ CONE, A( K, 1 ), LDA, B( K, 1 ), LDB,
$ ONE, A, LDA )
CALL CHEMM( <span class="string">'Left'</span>, UPLO, KB, K-1, HALF, A( K, K ),
$ LDA, B( K, 1 ), LDB, CONE, A( K, 1 ),
$ LDA )
CALL CTRMM( <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="CHEGS2.249"></a><a href="chegs2.f.html#CHEGS2.1">CHEGS2</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="CHEGST.257"></a><a href="chegst.f.html#CHEGST.1">CHEGST</a>
</span><span class="comment">*</span><span class="comment">
</span> END
</pre>
</body>
</html>
⌨️ 快捷键说明
复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?