3
4
5
6
7
8
9
10 INTEGER IAX, INCX, IX, JAX, JX, N
11 COMPLEX*16 ALPHA
12
13
14 INTEGER DESCX( * )
15 COMPLEX*16 TAU( * ), X( * )
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 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
146 $ LLD_, MB_, M_, NB_, N_, RSRC_
147 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
148 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
149 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
150 DOUBLE PRECISION ONE, ZERO
151 parameter( one = 1.0d+0, zero = 0.0d+0 )
152
153
154 INTEGER ICTXT, IIAX, INDXTAU, IXCOL, IXROW, J, JJAX,
155 $ KNT, MYCOL, MYROW, NPCOL, NPROW
156 DOUBLE PRECISION ALPHI, ALPHR, BETA, RSAFMN, SAFMIN, XNORM, ZR,
157 $ ZI
158
159
160 EXTERNAL blacs_gridinfo,
infog2l, pdznrm2,
161 $ zgebr2d, zgebs2d, pzscal,
162 $ pzdscal, dladiv
163
164
165 DOUBLE PRECISION DLAMCH, DLAPY3
167
168
169 INTRINSIC abs, dble, dcmplx, dimag, sign
170
171
172
173
174
175 ictxt = descx( ctxt_ )
176 CALL blacs_gridinfo( ictxt, nprow, npcol, myrow, mycol )
177
178 IF( incx.EQ.descx( m_ ) ) THEN
179
180
181
182 CALL infog2l( ix, jax, descx, nprow, npcol, myrow, mycol,
183 $ iiax, jjax, ixrow, ixcol )
184
185 IF( myrow.NE.ixrow )
186 $ RETURN
187
188
189
190 IF( mycol.EQ.ixcol ) THEN
191 j = iiax+(jjax-1)*descx( lld_ )
192 CALL zgebs2d( ictxt, 'Rowwise', ' ', 1, 1, x( j ), 1 )
193 alpha = x( j )
194 ELSE
195 CALL zgebr2d( ictxt, 'Rowwise', ' ', 1, 1, alpha, 1,
196 $ myrow, ixcol )
197 END IF
198
199 indxtau = iiax
200
201 ELSE
202
203
204
205 CALL infog2l( iax, jx, descx, nprow, npcol, myrow, mycol,
206 $ iiax, jjax, ixrow, ixcol )
207
208 IF( mycol.NE.ixcol )
209 $ RETURN
210
211
212
213 IF( myrow.EQ.ixrow ) THEN
214 j = iiax+(jjax-1)*descx( lld_ )
215 CALL zgebs2d( ictxt, 'Columnwise', ' ', 1, 1, x( j ), 1 )
216 alpha = x( j )
217 ELSE
218 CALL zgebr2d( ictxt, 'Columnwise', ' ', 1, 1, alpha, 1,
219 $ ixrow, mycol )
220 END IF
221
222 indxtau = jjax
223
224 END IF
225
226 IF( n.LE.0 ) THEN
227 tau( indxtau ) = zero
228 RETURN
229 END IF
230
231 CALL pdznrm2( n-1, xnorm, x, ix, jx, descx, incx )
232 alphr = dble( alpha )
233 alphi = dimag( alpha )
234
235 IF( xnorm.EQ.zero .AND. alphi.EQ.zero ) THEN
236
237
238
239 tau( indxtau ) = zero
240
241 ELSE
242
243
244
245 beta = -sign( dlapy3( alphr, alphi, xnorm ), alphr )
247 rsafmn = one / safmin
248 IF( abs( beta ).LT.safmin ) THEN
249
250
251
252 knt = 0
253 10 CONTINUE
254 knt = knt + 1
255 CALL pzdscal( n-1, rsafmn, x, ix, jx, descx, incx )
256 beta = beta*rsafmn
257 alphi = alphi*rsafmn
258 alphr = alphr*rsafmn
259 IF( abs( beta ).LT.safmin )
260 $ GO TO 10
261
262
263
264 CALL pdznrm2( n-1, xnorm, x, ix, jx, descx, incx )
265 alpha = dcmplx( alphr, alphi )
266 beta = -sign( dlapy3( alphr, alphi, xnorm ), alphr )
267 tau( indxtau ) = dcmplx( ( beta-alphr ) / beta,
268 $ -alphi / beta )
269 CALL dladiv( one, zero, dble( alpha-beta ),
270 $ dimag( alpha-beta ), zr, zi )
271 alpha = dcmplx( zr, zi )
272 CALL pzscal( n-1, alpha, x, ix, jx, descx, incx )
273
274
275
276 alpha = beta
277 DO 20 j = 1, knt
278 alpha = alpha*safmin
279 20 CONTINUE
280 ELSE
281 tau( indxtau ) = dcmplx( ( beta-alphr ) / beta,
282 $ -alphi / beta )
283 CALL dladiv( one, zero, dble( alpha-beta ),
284 $ dimag( alpha-beta ), zr, zi )
285 alpha = dcmplx( zr, zi )
286 CALL pzscal( n-1, alpha, x, ix, jx, descx, incx )
287 alpha = beta
288 END IF
289 END IF
290
291 RETURN
292
293
294
subroutine infog2l(grindx, gcindx, desc, nprow, npcol, myrow, mycol, lrindx, lcindx, rsrc, csrc)