3
4
5
6
7
8
9
10 CHARACTER UPLO
11 INTEGER IA, INFO, JA, LWORK, N
12
13
14 INTEGER DESCA( * )
15 DOUBLE PRECISION A( * ), D( * ), E( * ), TAU( * ), WORK( * )
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
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
223 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
224 $ LLD_, MB_, M_, NB_, N_, RSRC_
225 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
226 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
227 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
228 DOUBLE PRECISION ONE
229 parameter( one = 1.0d+0 )
230
231
232 LOGICAL LQUERY, UPPER
233 CHARACTER COLCTOP, ROWCTOP
234 INTEGER I, IACOL, IAROW, ICOFFA, ICTXT, IINFO, IPW,
235 $ IROFFA, J, JB, JX, K, KK, LWMIN, MYCOL, MYROW,
236 $ NB, NP, NPCOL, NPROW, NQ
237
238
239 INTEGER DESCW( DLEN_ ), IDUM1( 2 ), IDUM2( 2 )
240
241
245
246
247 LOGICAL LSAME
248 INTEGER INDXG2L, INDXG2P, NUMROC
250
251
252 INTRINSIC dble, ichar,
max,
min, mod
253
254
255
256
257
258 ictxt = desca( ctxt_ )
259 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
260
261
262
263 info = 0
264 IF( nprow.EQ.-1 ) THEN
265 info = -(600+ctxt_)
266 ELSE
267 CALL chk1mat( n, 2, n, 2, ia, ja, desca, 6, info )
268 upper =
lsame( uplo,
'U' )
269 IF( info.EQ.0 ) THEN
270 nb = desca( nb_ )
271 iroffa = mod( ia-1, desca( mb_ ) )
272 icoffa = mod( ja-1, desca( nb_ ) )
273 iarow =
indxg2p( ia, nb, myrow, desca( rsrc_ ), nprow )
274 iacol =
indxg2p( ja, nb, mycol, desca( csrc_ ), npcol )
275 np =
numroc( n, nb, myrow, iarow, nprow )
276 nq =
max( 1,
numroc( n+ja-1, nb, mycol, desca( csrc_ ),
277 $ npcol ) )
278 lwmin =
max( (np+1)*nb, 3*nb )
279
280 work( 1 ) = dble( lwmin )
281 lquery = ( lwork.EQ.-1 )
282 IF( .NOT.upper .AND. .NOT.
lsame( uplo,
'L' ) )
THEN
283 info = -1
284 ELSE IF( iroffa.NE.icoffa .OR. icoffa.NE.0 ) THEN
285 info = -5
286 ELSE IF( desca( mb_ ).NE.desca( nb_ ) ) THEN
287 info = -(600+nb_)
288 ELSE IF( lwork.LT.lwmin .AND. .NOT.lquery ) THEN
289 info = -11
290 END IF
291 END IF
292 IF( upper ) THEN
293 idum1( 1 ) = ichar( 'U' )
294 ELSE
295 idum1( 1 ) = ichar( 'L' )
296 END IF
297 idum2( 1 ) = 1
298 IF( lwork.EQ.-1 ) THEN
299 idum1( 2 ) = -1
300 ELSE
301 idum1( 2 ) = 1
302 END IF
303 idum2( 2 ) = 11
304 CALL pchk1mat( n, 2, n, 2, ia, ja, desca, 6, 2, idum1, idum2,
305 $ info )
306 END IF
307
308 IF( info.NE.0 ) THEN
309 CALL pxerbla( ictxt,
'PDSYTRD', -info )
310 RETURN
311 ELSE IF( lquery ) THEN
312 RETURN
313 END IF
314
315
316
317 IF( n.EQ.0 )
318 $ RETURN
319
320 CALL pb_topget( ictxt, 'Combine', 'Columnwise', colctop )
321 CALL pb_topget( ictxt, 'Combine', 'Rowwise', rowctop )
322 CALL pb_topset( ictxt, 'Combine', 'Columnwise', '1-tree' )
323 CALL pb_topset( ictxt, 'Combine', 'Rowwise', '1-tree' )
324
325 ipw = np * nb + 1
326
327 IF( upper ) THEN
328
329
330
331 kk = mod( ja+n-1, nb )
332 IF( kk.EQ.0 )
333 $ kk = nb
335 $ nb, mycol, desca( csrc_ ), npcol ), ictxt,
337
338 DO 10 k = n-kk+1, nb+1, -nb
339 jb =
min( n-k+1, nb )
340 i = ia + k - 1
341 j = ja + k - 1
342
343
344
345
346
347 CALL pdlatrd( uplo, k+jb-1, jb, a, ia, ja, desca, d, e, tau,
348 $ work, 1, 1, descw, work( ipw ) )
349
350
351
352
353
354 CALL pdsyr2k( uplo, 'No transpose', k-1, jb, -one, a, ia, j,
355 $ desca, work, 1, 1, descw, one, a, ia, ja,
356 $ desca )
357
358
359
360 jx =
min(
indxg2l( j, nb, 0, iacol, npcol ), nq )
361 CALL pdelset( a, i-1, j, desca, e( jx ) )
362
363 descw( csrc_ ) = mod( descw( csrc_ ) + npcol - 1, npcol )
364
365 10 CONTINUE
366
367
368
369 CALL pdsytd2( uplo,
min( n, nb ), a, ia, ja, desca, d, e,
370 $ tau, work, lwork, iinfo )
371
372 ELSE
373
374
375
376 kk = mod( ja+n-1, nb )
377 IF( kk.EQ.0 )
378 $ kk = nb
379 CALL descset( descw, n, nb, nb, nb, iarow, iacol, ictxt,
381
382 DO 20 k = 1, n-nb, nb
383 i = ia + k - 1
384 j = ja + k - 1
385
386
387
388
389
390 CALL pdlatrd( uplo, n-k+1, nb, a, i, j, desca, d, e, tau,
391 $ work, k, 1, descw, work( ipw ) )
392
393
394
395
396
397 CALL pdsyr2k( uplo, 'No transpose', n-k-nb+1, nb, -one, a,
398 $ i+nb, j, desca, work, k+nb, 1, descw, one, a,
399 $ i+nb, j+nb, desca )
400
401
402
403 jx =
min(
indxg2l( j+nb-1, nb, 0, iacol, npcol ), nq )
404 CALL pdelset( a, i+nb, j+nb-1, desca, e( jx ) )
405
406 descw( csrc_ ) = mod( descw( csrc_ ) + 1, npcol )
407
408 20 CONTINUE
409
410
411
412 CALL pdsytd2( uplo, kk, a, ia+k-1, ja+k-1, desca, d, e,
413 $ tau, work, lwork, iinfo )
414 END IF
415
416 CALL pb_topset( ictxt, 'Combine', 'Columnwise', colctop )
417 CALL pb_topset( ictxt, 'Combine', 'Rowwise', rowctop )
418
419 work( 1 ) = dble( lwmin )
420
421 RETURN
422
423
424
subroutine chk1mat(ma, mapos0, na, napos0, ia, ja, desca, descapos0, info)
subroutine descset(desc, m, n, mb, nb, irsrc, icsrc, ictxt, lld)
integer function indxg2l(indxglob, nb, iproc, isrcproc, nprocs)
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 pdelset(a, ia, ja, desca, alpha)
subroutine pdlatrd(uplo, n, nb, a, ia, ja, desca, d, e, tau, w, iw, jw, descw, work)
subroutine pdsytd2(uplo, n, a, ia, ja, desca, d, e, tau, work, lwork, info)
subroutine pxerbla(ictxt, srname, info)