3
4
5
6
7
8
9
10 INTEGER IA, INFO, JA, LWORK, M, N
11
12
13 INTEGER DESCA( * )
14 REAL A( * ), TAU( * ), WORK( * )
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
166 $ LLD_, MB_, M_, NB_, N_, RSRC_
167 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
168 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
169 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
170
171
172 LOGICAL LQUERY
173 CHARACTER COLBTOP, ROWBTOP
174 INTEGER IACOL, IAROW, IINFO, ICTXT, IPW, J, JB, JL, JN,
175 $ K, LWMIN, MP0, MU, MYCOL, MYROW, NPCOL, NPROW,
176 $ NQ0, NU
177
178
179 INTEGER IDUM1( 1 ), IDUM2( 1 )
180
181
185
186
187 INTEGER ICEIL, INDXG2P, NUMROC
189
190
191 INTRINSIC min, mod, real
192
193
194
195
196
197 ictxt = desca( ctxt_ )
198 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
199
200
201
202 info = 0
203 IF( nprow.EQ.-1 ) THEN
204 info = -(600+ctxt_)
205 ELSE
206 CALL chk1mat( m, 1, n, 2, ia, ja, desca, 6, info )
207 IF( info.EQ.0 ) THEN
208 iarow =
indxg2p( ia, desca( mb_ ), myrow, desca( rsrc_ ),
209 $ nprow )
210 iacol =
indxg2p( ja, desca( nb_ ), mycol, desca( csrc_ ),
211 $ npcol )
212 mp0 =
numroc( m+mod( ia-1, desca( mb_ ) ), desca( mb_ ),
213 $ myrow, iarow, nprow )
214 nq0 =
numroc( n+mod( ja-1, desca( nb_ ) ), desca( nb_ ),
215 $ mycol, iacol, npcol )
216 lwmin = desca( nb_ ) * ( mp0 + nq0 + desca( nb_ ) )
217
218 work( 1 ) = real( lwmin )
219 lquery = ( lwork.EQ.-1 )
220 IF( lwork.LT.lwmin .AND. .NOT.lquery )
221 $ info = -9
222 END IF
223 IF( lwork.EQ.-1 ) THEN
224 idum1( 1 ) = -1
225 ELSE
226 idum1( 1 ) = 1
227 END IF
228 idum2( 1 ) = 9
229 CALL pchk1mat( m, 1, n, 2, ia, ja, desca, 6, 1, idum1, idum2,
230 $ info )
231 END IF
232
233 IF( info.NE.0 ) THEN
234 CALL pxerbla( ictxt,
'PSGEQLF', -info )
235 RETURN
236 ELSE IF( lquery ) THEN
237 RETURN
238 END IF
239
240
241
242 IF( m.EQ.0 .OR. n.EQ.0 )
243 $ RETURN
244
246 ipw = desca( nb_ ) * desca( nb_ ) + 1
247 jn =
min(
iceil( ja+n-k, desca( nb_ ) ) * desca( nb_ ), ja+n-1 )
248 jl =
max( ( (ja+n-2) / desca( nb_ ) ) * desca( nb_ ) + 1, ja )
249 CALL pb_topget( ictxt, 'Broadcast', 'Rowwise', rowbtop )
250 CALL pb_topget( ictxt, 'Broadcast', 'Columnwise', colbtop )
251 CALL pb_topset( ictxt, 'Broadcast', 'Rowwise', 'D-ring' )
252 CALL pb_topset( ictxt, 'Broadcast', 'Columnwise', ' ' )
253
254 IF( jl.GE.jn+1 ) THEN
255
256
257
258 DO 10 j = jl, jn+1, -desca( nb_ )
259 jb =
min( ja+n-j, desca( nb_ ) )
260
261
262
263
264 CALL psgeql2( m-n+j+jb-ja, jb, a, ia, j, desca, tau, work,
265 $ lwork, iinfo )
266
267 IF( j.GT.ja ) THEN
268
269
270
271
272 CALL pslarft(
'Backward',
'Columnwise', m-n+j+jb-ja, jb,
273 $ a, ia, j, desca, tau, work, work( ipw ) )
274
275
276
277
278 CALL pslarfb(
'Left',
'Transpose',
'Backward',
279 $ 'Columnwise', m-n+j+jb-ja, j-ja, jb, a, ia,
280 $ j, desca, work, a, ia, ja, desca,
281 $ work( ipw ) )
282 END IF
283
284 10 CONTINUE
285
286 mu = m - n + jn - ja + 1
287 nu = jn - ja + 1
288
289 ELSE
290
291 mu = m
292 nu = n
293
294 END IF
295
296
297
298 IF( mu.GT.0 .AND. nu.GT.0 )
299 $
CALL psgeql2( mu, nu, a, ia, ja, desca, tau, work, lwork,
300 $ iinfo )
301
302 CALL pb_topset( ictxt, 'Broadcast', 'Rowwise', rowbtop )
303 CALL pb_topset( ictxt, 'Broadcast', 'Columnwise', colbtop )
304
305 work( 1 ) = real( lwmin )
306
307 RETURN
308
309
310
subroutine chk1mat(ma, mapos0, na, napos0, ia, ja, desca, descapos0, info)
integer function iceil(inum, idenom)
integer function indxg2p(indxglob, nb, iproc, isrcproc, nprocs)
integer function numroc(n, nb, iproc, isrcproc, nprocs)
subroutine pchk1mat(ma, mapos0, na, napos0, ia, ja, desca, descapos0, nextra, ex, expos, info)
subroutine psgeql2(m, n, a, ia, ja, desca, tau, work, lwork, info)
subroutine pslarfb(side, trans, direct, storev, m, n, k, v, iv, jv, descv, t, c, ic, jc, descc, work)
subroutine pslarft(direct, storev, n, k, v, iv, jv, descv, tau, t, work)
subroutine pxerbla(ictxt, srname, info)