LAPACK 3.12.1
LAPACK: Linear Algebra PACKage
Loading...
Searching...
No Matches
cgeqr.f
Go to the documentation of this file.
1*> \brief \b CGEQR
2*
3* Definition:
4* ===========
5*
6* SUBROUTINE CGEQR( M, N, A, LDA, T, TSIZE, WORK, LWORK,
7* INFO )
8*
9* .. Scalar Arguments ..
10* INTEGER INFO, LDA, M, N, TSIZE, LWORK
11* ..
12* .. Array Arguments ..
13* COMPLEX A( LDA, * ), T( * ), WORK( * )
14* ..
15*
16*
17*> \par Purpose:
18* =============
19*>
20*> \verbatim
21*>
22*> CGEQR computes a QR factorization of a complex M-by-N matrix A:
23*>
24*> A = Q * ( R ),
25*> ( 0 )
26*>
27*> where:
28*>
29*> Q is a M-by-M orthogonal matrix;
30*> R is an upper-triangular N-by-N matrix;
31*> 0 is a (M-N)-by-N zero matrix, if M > N.
32*>
33*> \endverbatim
34*
35* Arguments:
36* ==========
37*
38*> \param[in] M
39*> \verbatim
40*> M is INTEGER
41*> The number of rows of the matrix A. M >= 0.
42*> \endverbatim
43*>
44*> \param[in] N
45*> \verbatim
46*> N is INTEGER
47*> The number of columns of the matrix A. N >= 0.
48*> \endverbatim
49*>
50*> \param[in,out] A
51*> \verbatim
52*> A is COMPLEX array, dimension (LDA,N)
53*> On entry, the M-by-N matrix A.
54*> On exit, the elements on and above the diagonal of the array
55*> contain the min(M,N)-by-N upper trapezoidal matrix R
56*> (R is upper triangular if M >= N);
57*> the elements below the diagonal are used to store part of the
58*> data structure to represent Q.
59*> \endverbatim
60*>
61*> \param[in] LDA
62*> \verbatim
63*> LDA is INTEGER
64*> The leading dimension of the array A. LDA >= max(1,M).
65*> \endverbatim
66*>
67*> \param[out] T
68*> \verbatim
69*> T is COMPLEX array, dimension (MAX(5,TSIZE))
70*> On exit, if INFO = 0, T(1) returns optimal (or either minimal
71*> or optimal, if query is assumed) TSIZE. See TSIZE for details.
72*> Remaining T contains part of the data structure used to represent Q.
73*> If one wants to apply or construct Q, then one needs to keep T
74*> (in addition to A) and pass it to further subroutines.
75*> \endverbatim
76*>
77*> \param[in] TSIZE
78*> \verbatim
79*> TSIZE is INTEGER
80*> If TSIZE >= 5, the dimension of the array T.
81*> If TSIZE = -1 or -2, then a workspace query is assumed. The routine
82*> only calculates the sizes of the T and WORK arrays, returns these
83*> values as the first entries of the T and WORK arrays, and no error
84*> message related to T or WORK is issued by XERBLA.
85*> If TSIZE = -1, the routine calculates optimal size of T for the
86*> optimum performance and returns this value in T(1).
87*> If TSIZE = -2, the routine calculates minimal size of T and
88*> returns this value in T(1).
89*> \endverbatim
90*>
91*> \param[out] WORK
92*> \verbatim
93*> (workspace) COMPLEX array, dimension (MAX(1,LWORK))
94*> On exit, if INFO = 0, WORK(1) contains optimal (or either minimal
95*> or optimal, if query was assumed) LWORK.
96*> See LWORK for details.
97*> \endverbatim
98*>
99*> \param[in] LWORK
100*> \verbatim
101*> LWORK is INTEGER
102*> The dimension of the array WORK. LWORK >= 1.
103*> If LWORK = -1 or -2, then a workspace query is assumed. The routine
104*> only calculates the sizes of the T and WORK arrays, returns these
105*> values as the first entries of the T and WORK arrays, and no error
106*> message related to T or WORK is issued by XERBLA.
107*> If LWORK = -1, the routine calculates optimal size of WORK for the
108*> optimal performance and returns this value in WORK(1).
109*> If LWORK = -2, the routine calculates minimal size of WORK and
110*> returns this value in WORK(1).
111*> \endverbatim
112*>
113*> \param[out] INFO
114*> \verbatim
115*> INFO is INTEGER
116*> = 0: successful exit
117*> < 0: if INFO = -i, the i-th argument had an illegal value
118*> \endverbatim
119*
120* Authors:
121* ========
122*
123*> \author Univ. of Tennessee
124*> \author Univ. of California Berkeley
125*> \author Univ. of Colorado Denver
126*> \author NAG Ltd.
127*
128*> \par Further Details
129* ====================
130*>
131*> \verbatim
132*>
133*> The goal of the interface is to give maximum freedom to the developers for
134*> creating any QR factorization algorithm they wish. The triangular
135*> (trapezoidal) R has to be stored in the upper part of A. The lower part of A
136*> and the array T can be used to store any relevant information for applying or
137*> constructing the Q factor. The WORK array can safely be discarded after exit.
138*>
139*> Caution: One should not expect the sizes of T and WORK to be the same from one
140*> LAPACK implementation to the other, or even from one execution to the other.
141*> A workspace query (for T and WORK) is needed at each execution. However,
142*> for a given execution, the size of T and WORK are fixed and will not change
143*> from one query to the next.
144*>
145*> \endverbatim
146*>
147*> \par Further Details particular to this LAPACK implementation:
148* ==============================================================
149*>
150*> \verbatim
151*>
152*> These details are particular for this LAPACK implementation. Users should not
153*> take them for granted. These details may change in the future, and are not likely
154*> true for another LAPACK implementation. These details are relevant if one wants
155*> to try to understand the code. They are not part of the interface.
156*>
157*> In this version,
158*>
159*> T(2): row block size (MB)
160*> T(3): column block size (NB)
161*> T(6:TSIZE): data structure needed for Q, computed by
162*> CLATSQR or CGEQRT
163*>
164*> Depending on the matrix dimensions M and N, and row and column
165*> block sizes MB and NB returned by ILAENV, CGEQR will use either
166*> CLATSQR (if the matrix is tall-and-skinny) or CGEQRT to compute
167*> the QR factorization.
168*>
169*> \endverbatim
170*>
171*> \ingroup geqr
172*>
173* =====================================================================
174 SUBROUTINE cgeqr( M, N, A, LDA, T, TSIZE, WORK, LWORK,
175 $ INFO )
176*
177* -- LAPACK computational routine --
178* -- LAPACK is a software package provided by Univ. of Tennessee, --
179* -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd. --
180*
181* .. Scalar Arguments ..
182 INTEGER INFO, LDA, M, N, TSIZE, LWORK
183* ..
184* .. Array Arguments ..
185 COMPLEX A( LDA, * ), T( * ), WORK( * )
186* ..
187*
188* =====================================================================
189*
190* ..
191* .. Local Scalars ..
192 LOGICAL LQUERY, LMINWS, MINT, MINW
193 INTEGER MB, NB, MINTSZ, NBLCKS, LWMIN, LWREQ
194* ..
195* .. External Functions ..
196 LOGICAL LSAME
197 REAL SROUNDUP_LWORK
198 EXTERNAL lsame, sroundup_lwork
199* ..
200* .. External Subroutines ..
201 EXTERNAL clatsqr, cgeqrt, xerbla
202* ..
203* .. Intrinsic Functions ..
204 INTRINSIC max, min, mod
205* ..
206* .. External Functions ..
207 INTEGER ILAENV
208 EXTERNAL ilaenv
209* ..
210* .. Executable Statements ..
211*
212* Test the input arguments
213*
214 info = 0
215*
216 lquery = ( tsize.EQ.-1 .OR. tsize.EQ.-2 .OR.
217 $ lwork.EQ.-1 .OR. lwork.EQ.-2 )
218*
219 mint = .false.
220 minw = .false.
221 IF( tsize.EQ.-2 .OR. lwork.EQ.-2 ) THEN
222 IF( tsize.NE.-1 ) mint = .true.
223 IF( lwork.NE.-1 ) minw = .true.
224 END IF
225*
226* Determine the block size
227*
228 IF( min( m, n ).GT.0 ) THEN
229 mb = ilaenv( 1, 'CGEQR ', ' ', m, n, 1, -1 )
230 nb = ilaenv( 1, 'CGEQR ', ' ', m, n, 2, -1 )
231 ELSE
232 mb = m
233 nb = 1
234 END IF
235 IF( mb.GT.m .OR. mb.LE.n ) mb = m
236 IF( nb.GT.min( m, n ) .OR. nb.LT.1 ) nb = 1
237 mintsz = n + 5
238 IF( mb.GT.n .AND. m.GT.n ) THEN
239 IF( mod( m - n, mb - n ).EQ.0 ) THEN
240 nblcks = ( m - n ) / ( mb - n )
241 ELSE
242 nblcks = ( m - n ) / ( mb - n ) + 1
243 END IF
244 ELSE
245 nblcks = 1
246 END IF
247*
248* Determine if the workspace size satisfies minimal size
249*
250 lwmin = max( 1, n )
251 lwreq = max( 1, n*nb )
252 lminws = .false.
253 IF( ( tsize.LT.max( 1, nb*n*nblcks + 5 ) .OR. lwork.LT.lwreq )
254 $ .AND. ( lwork.GE.n ) .AND. ( tsize.GE.mintsz )
255 $ .AND. ( .NOT.lquery ) ) THEN
256 IF( tsize.LT.max( 1, nb*n*nblcks + 5 ) ) THEN
257 lminws = .true.
258 nb = 1
259 mb = m
260 END IF
261 IF( lwork.LT.lwreq ) THEN
262 lminws = .true.
263 nb = 1
264 END IF
265 END IF
266*
267 IF( m.LT.0 ) THEN
268 info = -1
269 ELSE IF( n.LT.0 ) THEN
270 info = -2
271 ELSE IF( lda.LT.max( 1, m ) ) THEN
272 info = -4
273 ELSE IF( tsize.LT.max( 1, nb*n*nblcks + 5 )
274 $ .AND. ( .NOT.lquery ) .AND. ( .NOT.lminws ) ) THEN
275 info = -6
276 ELSE IF( ( lwork.LT.lwreq ) .AND. ( .NOT.lquery )
277 $ .AND. ( .NOT.lminws ) ) THEN
278 info = -8
279 END IF
280*
281 IF( info.EQ.0 ) THEN
282 IF( mint ) THEN
283 t( 1 ) = cmplx( mintsz )
284 ELSE
285 t( 1 ) = cmplx( nb*n*nblcks + 5 )
286 END IF
287 t( 2 ) = cmplx( mb )
288 t( 3 ) = cmplx( nb )
289 IF( minw ) THEN
290 work( 1 ) = sroundup_lwork( lwmin )
291 ELSE
292 work( 1 ) = sroundup_lwork( lwreq )
293 END IF
294 END IF
295 IF( info.NE.0 ) THEN
296 CALL xerbla( 'CGEQR', -info )
297 RETURN
298 ELSE IF( lquery ) THEN
299 RETURN
300 END IF
301*
302* Quick return if possible
303*
304 IF( min( m, n ).EQ.0 ) THEN
305 RETURN
306 END IF
307*
308* The QR Decomposition
309*
310 IF( ( m.LE.n ) .OR. ( mb.LE.n ) .OR. ( mb.GE.m ) ) THEN
311 CALL cgeqrt( m, n, nb, a, lda, t( 6 ), nb, work, info )
312 ELSE
313 CALL clatsqr( m, n, mb, nb, a, lda, t( 6 ), nb, work,
314 $ lwork, info )
315 END IF
316*
317 work( 1 ) = sroundup_lwork( lwreq )
318*
319 RETURN
320*
321* End of CGEQR
322*
323 END
subroutine xerbla(srname, info)
Definition cblat2.f:3285
subroutine cgeqr(m, n, a, lda, t, tsize, work, lwork, info)
CGEQR
Definition cgeqr.f:176
subroutine cgeqrt(m, n, nb, a, lda, t, ldt, work, info)
CGEQRT
Definition cgeqrt.f:139
subroutine clatsqr(m, n, mb, nb, a, lda, t, ldt, work, lwork, info)
CLATSQR
Definition clatsqr.f:172