3
4
5
6
7
8
9
10 CHARACTER UPLO
11 INTEGER IA, IW, JA, JW, N, NB
12
13
14 INTEGER DESCA( * ), DESCW( * )
15 DOUBLE PRECISION A( * ), D( * ), E( * ), TAU( * ), W( * ),
16 $ WORK( * )
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
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
223 $ LLD_, MB_, M_, NB_, N_, RSRC_
224 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
225 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
226 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
227 DOUBLE PRECISION HALF, ONE, ZERO
228 parameter( half = 0.5d+0, one = 1.0d+0, zero = 0.0d+0 )
229
230
231 INTEGER I, IACOL, IAROW, ICTXT, II, J, JJ, JP, JWK, K,
232 $ KW, MYCOL, MYROW, NPCOL, NPROW, NQ
233 DOUBLE PRECISION ALPHA
234
235
236 INTEGER DESCD( DLEN_ ), DESCE( DLEN_ ), DESCWK( DLEN_ )
237
238
239 EXTERNAL blacs_gridinfo,
descset, dgebr2d, dgebs2d,
242 $ pdsymv
243
244
245 LOGICAL LSAME
246 INTEGER NUMROC
248
249
251
252
253
254
255
256 IF( n.LE.0 )
257 $ RETURN
258
259 ictxt = desca( ctxt_ )
260 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
261 nq =
max( 1,
numroc( ja+n-1, desca( nb_ ), mycol, desca( csrc_ ),
262 $ npcol ) )
263 CALL descset( descd, 1, ja+n-1, 1, desca( nb_ ), myrow,
264 $ desca( csrc_ ), desca( ctxt_ ), 1 )
265
266 IF(
lsame( uplo,
'U' ) )
THEN
267
268 CALL infog2l( n+ia-nb, n+ja-nb, desca, nprow, npcol, myrow,
269 $ mycol, ii, jj, iarow, iacol )
270 CALL descset( descwk, 1, descw( nb_ ), 1, descw( nb_ ), iarow,
271 $ iacol, ictxt, 1 )
272 CALL descset( desce, 1, ja+n-1, 1, desca( nb_ ), myrow,
273 $ desca( csrc_ ), desca( ctxt_ ), 1 )
274
275
276
277 DO 10 j = ja+n-1, ja+n-nb, -1
278 i = ia + j - ja
279 k = j - ja + 1
280 kw = mod( k-1, desca( mb_ ) ) + 1
281
282
283
284 CALL pdgemv( 'No transpose', k, n-k, -one, a, ia, j+1,
285 $ desca, w, iw+k-1, jw+kw, descw, descw( m_ ),
286 $ one, a, ia, j, desca, 1 )
287 CALL pdgemv( 'No transpose', k, n-k, -one, w, iw, jw+kw,
288 $ descw, a, i, j+1, desca, desca( m_ ), one, a,
289 $ ia, j, desca, 1 )
290 IF( n-k.GT.0 )
291 $
CALL pdelset( a, i, j+1, desca, e( jp ) )
292
293
294
295
296 jp =
min( jj+kw-1, nq )
297 CALL pdlarfg( k-1, e( jp ), i-1, j, a, ia, j, desca, 1,
298 $ tau )
299 CALL pdelset( a, i-1, j, desca, one )
300
301
302
303 CALL pdsymv( 'Upper', k-1, one, a, ia, ja, desca, a, ia, j,
304 $ desca, 1, zero, w, iw, jw+kw-1, descw, 1 )
305
306 jwk = mod( k-1, descwk( nb_ ) ) + 2
307 CALL pdgemv( 'Transpose', k-1, n-k, one, w, iw, jw+kw,
308 $ descw, a, ia, j, desca, 1, zero, work, 1, jwk,
309 $ descwk, descwk( m_ ) )
310 CALL pdgemv( 'No transpose', k-1, n-k, -one, a, ia, j+1,
311 $ desca, work, 1, jwk, descwk, descwk( m_ ), one,
312 $ w, iw, jw+kw-1, descw, 1 )
313 CALL pdgemv( 'Transpose', k-1, n-k, one, a, ia, j+1, desca,
314 $ a, ia, j, desca, 1, zero, work, 1, jwk, descwk,
315 $ descwk( m_ ) )
316 CALL pdgemv( 'No transpose', k-1, n-k, -one, w, iw, jw+kw,
317 $ descw, work, 1, jwk, descwk, descwk( m_ ), one,
318 $ w, iw, jw+kw-1, descw, 1 )
319 CALL pdscal( k-1, tau( jp ), w, iw, jw+kw-1, descw, 1 )
320
321 CALL pddot( k-1, alpha, w, iw, jw+kw-1, descw, 1, a, ia, j,
322 $ desca, 1 )
323 IF( mycol.EQ.iacol )
324 $ alpha = -half*tau( jp )*alpha
325 CALL pdaxpy( k-1, alpha, a, ia, j, desca, 1, w, iw, jw+kw-1,
326 $ descw, 1 )
327 IF( mycol.EQ.iacol ) THEN
328 CALL pdelget(
'E',
' ', d( jp ), a, i, j, desca )
329 END IF
330
331 10 CONTINUE
332
333 ELSE
334
335 CALL infog2l( ia, ja, desca, nprow, npcol, myrow, mycol, ii,
336 $ jj, iarow, iacol )
337 CALL descset( descwk, 1, descw( nb_ ), 1, descw( nb_ ), iarow,
338 $ iacol, ictxt, 1 )
339 CALL descset( desce, 1, ja+n-2, 1, desca( nb_ ), myrow,
340 $ desca( csrc_ ), desca( ctxt_ ), 1 )
341
342
343
344 DO 20 j = ja, ja+nb-1
345 i = ia + j - ja
346 k = j - ja + 1
347
348
349
350 CALL pdgemv( 'No transpose', n-k+1, k-1, -one, a, i, ja,
351 $ desca, w, iw+k-1, jw, descw, descw( m_ ), one,
352 $ a, i, j, desca, 1 )
353 CALL pdgemv( 'No transpose', n-k+1, k-1, -one, w, iw+k-1,
354 $ jw, descw, a, i, ja, desca, desca( m_ ), one,
355 $ a, i, j, desca, 1 )
356 IF( k.GT.1 )
357 $
CALL pdelset( a, i, j-1, desca, e( jp ) )
358
359
360
361
362
363 jp =
min( jj+k-1, nq )
364 CALL pdlarfg( n-k, e( jp ), i+1, j, a, i+2, j, desca, 1,
365 $ tau )
366 CALL pdelset( a, i+1, j, desca, one )
367
368
369
370 CALL pdsymv( 'Lower', n-k, one, a, i+1, j+1, desca, a, i+1,
371 $ j, desca, 1, zero, w, iw+k, jw+k-1, descw, 1 )
372
373 CALL pdgemv( 'Transpose', n-k, k-1, one, w, iw+k, jw, descw,
374 $ a, i+1, j, desca, 1, zero, work, 1, 1, descwk,
375 $ descwk( m_ ) )
376 CALL pdgemv( 'No transpose', n-k, k-1, -one, a, i+1, ja,
377 $ desca, work, 1, 1, descwk, descwk( m_ ), one, w,
378 $ iw+k, jw+k-1, descw, 1 )
379 CALL pdgemv( 'Transpose', n-k, k-1, one, a, i+1, ja, desca,
380 $ a, i+1, j, desca, 1, zero, work, 1, 1, descwk,
381 $ descwk( m_ ) )
382 CALL pdgemv( 'No transpose', n-k, k-1, -one, w, iw+k, jw,
383 $ descw, work, 1, 1, descwk, descwk( m_ ), one, w,
384 $ iw+k, jw+k-1, descw, 1 )
385 CALL pdscal( n-k, tau( jp ), w, iw+k, jw+k-1, descw, 1 )
386 CALL pddot( n-k, alpha, w, iw+k, jw+k-1, descw, 1, a, i+1,
387 $ j, desca, 1 )
388 IF( mycol.EQ.iacol )
389 $ alpha = -half*tau( jp )*alpha
390 CALL pdaxpy( n-k, alpha, a, i+1, j, desca, 1, w, iw+k,
391 $ jw+k-1, descw, 1 )
392 IF( mycol.EQ.iacol ) THEN
393 CALL pdelget(
'E',
' ', d( jp ), a, i, j, desca )
394 END IF
395
396 20 CONTINUE
397
398 END IF
399
400
401
402 IF( mycol.EQ.iacol ) THEN
403 IF( myrow.EQ.iarow ) THEN
404 CALL dgebs2d( ictxt, 'Columnwise', ' ', 1, nb, d( jj ), 1 )
405 ELSE
406 CALL dgebr2d( ictxt, 'Columnwise', ' ', 1, nb, d( jj ), 1,
407 $ iarow, mycol )
408 END IF
409 END IF
410
411 RETURN
412
413
414
subroutine descset(desc, m, n, mb, nb, irsrc, icsrc, ictxt, lld)
subroutine infog2l(grindx, gcindx, desc, nprow, npcol, myrow, mycol, lrindx, lcindx, rsrc, csrc)
integer function numroc(n, nb, iproc, isrcproc, nprocs)
subroutine pdelget(scope, top, alpha, a, ia, ja, desca)
subroutine pdelset(a, ia, ja, desca, alpha)
subroutine pdlarfg(n, alpha, iax, jax, x, ix, jx, descx, incx, tau)