You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

csysv_aa.f 7.9 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249
  1. *> \brief <b> CSYSV_AA computes the solution to system of linear equations A * X = B for SY 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 CSYSV_AA + dependencies
  10. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.tgz?format=tgz&filename=/lapack/lapack_routine/csysv_aa.f">
  11. *> [TGZ]</a>
  12. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.zip?format=zip&filename=/lapack/lapack_routine/csysv_aa.f">
  13. *> [ZIP]</a>
  14. *> <a href="http://www.netlib.org/cgi-bin/netlibfiles.txt?format=txt&filename=/lapack/lapack_routine/csysv_aa.f">
  15. *> [TXT]</a>
  16. *> \endhtmlonly
  17. *
  18. * Definition:
  19. * ===========
  20. *
  21. * SUBROUTINE CSYSV_AA( UPLO, N, NRHS, A, LDA, IPIV, B, LDB, WORK,
  22. * LWORK, INFO )
  23. *
  24. * .. Scalar Arguments ..
  25. * CHARACTER UPLO
  26. * INTEGER N, NRHS, LDA, LDB, LWORK, INFO
  27. * ..
  28. * .. Array Arguments ..
  29. * INTEGER IPIV( * )
  30. * COMPLEX A( LDA, * ), B( LDB, * ), WORK( * )
  31. * ..
  32. *
  33. *
  34. *> \par Purpose:
  35. * =============
  36. *>
  37. *> \verbatim
  38. *>
  39. *> CSYSV computes the solution to a complex system of linear equations
  40. *> A * X = B,
  41. *> where A is an N-by-N symmetric matrix and X and B are N-by-NRHS
  42. *> matrices.
  43. *>
  44. *> Aasen's algorithm is used to factor A as
  45. *> A = U**T * T * U, if UPLO = 'U', or
  46. *> A = L * T * L**T, if UPLO = 'L',
  47. *> where U (or L) is a product of permutation and unit upper (lower)
  48. *> triangular matrices, and T is symmetric tridiagonal. The factored
  49. *> form of A is then used to solve the system of equations A * X = B.
  50. *> \endverbatim
  51. *
  52. * Arguments:
  53. * ==========
  54. *
  55. *> \param[in] UPLO
  56. *> \verbatim
  57. *> UPLO is CHARACTER*1
  58. *> = 'U': Upper triangle of A is stored;
  59. *> = 'L': Lower triangle of A is stored.
  60. *> \endverbatim
  61. *>
  62. *> \param[in] N
  63. *> \verbatim
  64. *> N is INTEGER
  65. *> The number of linear equations, i.e., the order of the
  66. *> matrix A. N >= 0.
  67. *> \endverbatim
  68. *>
  69. *> \param[in] NRHS
  70. *> \verbatim
  71. *> NRHS is INTEGER
  72. *> The number of right hand sides, i.e., the number of columns
  73. *> of the matrix B. NRHS >= 0.
  74. *> \endverbatim
  75. *>
  76. *> \param[in,out] A
  77. *> \verbatim
  78. *> A is COMPLEX array, dimension (LDA,N)
  79. *> On entry, the symmetric matrix A. If UPLO = 'U', the leading
  80. *> N-by-N upper triangular part of A contains the upper
  81. *> triangular part of the matrix A, and the strictly lower
  82. *> triangular part of A is not referenced. If UPLO = 'L', the
  83. *> leading N-by-N lower triangular part of A contains the lower
  84. *> triangular part of the matrix A, and the strictly upper
  85. *> triangular part of A is not referenced.
  86. *>
  87. *> On exit, if INFO = 0, the tridiagonal matrix T and the
  88. *> multipliers used to obtain the factor U or L from the
  89. *> factorization A = U**T*T*U or A = L*T*L**T as computed by
  90. *> CSYTRF.
  91. *> \endverbatim
  92. *>
  93. *> \param[in] LDA
  94. *> \verbatim
  95. *> LDA is INTEGER
  96. *> The leading dimension of the array A. LDA >= max(1,N).
  97. *> \endverbatim
  98. *>
  99. *> \param[out] IPIV
  100. *> \verbatim
  101. *> IPIV is INTEGER array, dimension (N)
  102. *> On exit, it contains the details of the interchanges, i.e.,
  103. *> the row and column k of A were interchanged with the
  104. *> row and column IPIV(k).
  105. *> \endverbatim
  106. *>
  107. *> \param[in,out] B
  108. *> \verbatim
  109. *> B is COMPLEX array, dimension (LDB,NRHS)
  110. *> On entry, the N-by-NRHS right hand side matrix B.
  111. *> On exit, if INFO = 0, the N-by-NRHS solution matrix X.
  112. *> \endverbatim
  113. *>
  114. *> \param[in] LDB
  115. *> \verbatim
  116. *> LDB is INTEGER
  117. *> The leading dimension of the array B. LDB >= max(1,N).
  118. *> \endverbatim
  119. *>
  120. *> \param[out] WORK
  121. *> \verbatim
  122. *> WORK is COMPLEX array, dimension (MAX(1,LWORK))
  123. *> On exit, if INFO = 0, WORK(1) returns the optimal LWORK.
  124. *> \endverbatim
  125. *>
  126. *> \param[in] LWORK
  127. *> \verbatim
  128. *> LWORK is INTEGER
  129. *> The length of WORK. LWORK >= MAX(2*N, 3*N-2), and for
  130. *> the best performance, LWORK >= max(1,N*NB), where NB is
  131. *> the optimal blocksize for CSYTRF_AA.
  132. *>
  133. *> If LWORK = -1, then a workspace query is assumed; the routine
  134. *> only calculates the optimal size of the WORK array, returns
  135. *> this value as the first entry of the WORK array, and no error
  136. *> message related to LWORK is issued by XERBLA.
  137. *> \endverbatim
  138. *>
  139. *> \param[out] INFO
  140. *> \verbatim
  141. *> INFO is INTEGER
  142. *> = 0: successful exit
  143. *> < 0: if INFO = -i, the i-th argument had an illegal value
  144. *> > 0: if INFO = i, D(i,i) is exactly zero. The factorization
  145. *> has been completed, but the block diagonal matrix D is
  146. *> exactly singular, so the solution could not be computed.
  147. *> \endverbatim
  148. *
  149. * Authors:
  150. * ========
  151. *
  152. *> \author Univ. of Tennessee
  153. *> \author Univ. of California Berkeley
  154. *> \author Univ. of Colorado Denver
  155. *> \author NAG Ltd.
  156. *
  157. *> \ingroup hesv_aa
  158. *
  159. * =====================================================================
  160. SUBROUTINE CSYSV_AA( UPLO, N, NRHS, A, LDA, IPIV, B, LDB, WORK,
  161. $ LWORK, INFO )
  162. *
  163. * -- LAPACK driver routine --
  164. * -- LAPACK is a software package provided by Univ. of Tennessee, --
  165. * -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
  166. *
  167. * .. Scalar Arguments ..
  168. CHARACTER UPLO
  169. INTEGER INFO, LDA, LDB, LWORK, N, NRHS
  170. * ..
  171. * .. Array Arguments ..
  172. INTEGER IPIV( * )
  173. COMPLEX A( LDA, * ), B( LDB, * ), WORK( * )
  174. * ..
  175. *
  176. * =====================================================================
  177. *
  178. * .. Local Scalars ..
  179. LOGICAL LQUERY
  180. INTEGER LWKOPT, LWKOPT_SYTRF, LWKOPT_SYTRS
  181. * ..
  182. * .. External Functions ..
  183. LOGICAL LSAME
  184. INTEGER ILAENV
  185. REAL SROUNDUP_LWORK
  186. EXTERNAL ILAENV, LSAME, SROUNDUP_LWORK
  187. * ..
  188. * .. External Subroutines ..
  189. EXTERNAL XERBLA, CSYTRF_AA, CSYTRS_AA
  190. * ..
  191. * .. Intrinsic Functions ..
  192. INTRINSIC MAX
  193. * ..
  194. * .. Executable Statements ..
  195. *
  196. * Test the input parameters.
  197. *
  198. INFO = 0
  199. LQUERY = ( LWORK.EQ.-1 )
  200. IF( .NOT.LSAME( UPLO, 'U' ) .AND. .NOT.LSAME( UPLO, 'L' ) ) THEN
  201. INFO = -1
  202. ELSE IF( N.LT.0 ) THEN
  203. INFO = -2
  204. ELSE IF( NRHS.LT.0 ) THEN
  205. INFO = -3
  206. ELSE IF( LDA.LT.MAX( 1, N ) ) THEN
  207. INFO = -5
  208. ELSE IF( LDB.LT.MAX( 1, N ) ) THEN
  209. INFO = -8
  210. ELSE IF( LWORK.LT.MAX(2*N, 3*N-2) .AND. .NOT.LQUERY ) THEN
  211. INFO = -10
  212. END IF
  213. *
  214. IF( INFO.EQ.0 ) THEN
  215. CALL CSYTRF_AA( UPLO, N, A, LDA, IPIV, WORK, -1, INFO )
  216. LWKOPT_SYTRF = INT( WORK(1) )
  217. CALL CSYTRS_AA( UPLO, N, NRHS, A, LDA, IPIV, B, LDB, WORK,
  218. $ -1, INFO )
  219. LWKOPT_SYTRS = INT( WORK(1) )
  220. LWKOPT = MAX( LWKOPT_SYTRF, LWKOPT_SYTRS )
  221. WORK( 1 ) = SROUNDUP_LWORK(LWKOPT)
  222. END IF
  223. *
  224. IF( INFO.NE.0 ) THEN
  225. CALL XERBLA( 'CSYSV_AA ', -INFO )
  226. RETURN
  227. ELSE IF( LQUERY ) THEN
  228. RETURN
  229. END IF
  230. *
  231. * Compute the factorization A = U**T*T*U or A = L*T*L**T.
  232. *
  233. CALL CSYTRF_AA( UPLO, N, A, LDA, IPIV, WORK, LWORK, INFO )
  234. IF( INFO.EQ.0 ) THEN
  235. *
  236. * Solve the system A*X = B, overwriting B with X.
  237. *
  238. CALL CSYTRS_AA( UPLO, N, NRHS, A, LDA, IPIV, B, LDB, WORK,
  239. $ LWORK, INFO )
  240. *
  241. END IF
  242. *
  243. WORK( 1 ) = SROUNDUP_LWORK(LWKOPT)
  244. *
  245. RETURN
  246. *
  247. * End of CSYSV_AA
  248. *
  249. END