2
3
4
5
6
7
8
9 INTEGER II, JJ, M
10 DOUBLE PRECISION H33, H43H34, H44
11
12
13 INTEGER DESCA( * )
14 DOUBLE PRECISION A( * ), V( * )
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 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
114 $ LLD_, MB_, M_, NB_, N_, RSRC_
115 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
116 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
117 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
118
119
120 INTEGER CONTXT, DOWN, HBL, ICOL, IROW, JSRC, LDA, LEFT,
121 $ MODKM1, MYCOL, MYROW, NPCOL, NPROW, NUM, RIGHT,
122 $ RSRC, UP
123 DOUBLE PRECISION H22, H33S, H44S, S, V1, V2
124
125
126 DOUBLE PRECISION BUF( 4 ), H11( 1 ), H12( 1 ), H21( 1 ), V3( 1 )
127
128
129 EXTERNAL blacs_gridinfo, dgerv2d, dgesd2d,
infog2l
130
131
132 INTRINSIC abs, mod
133
134
135
136 hbl = desca( mb_ )
137 contxt = desca( ctxt_ )
138 lda = desca( lld_ )
139 CALL blacs_gridinfo( contxt, nprow, npcol, myrow, mycol )
140 left = mod( mycol+npcol-1, npcol )
141 right = mod( mycol+1, npcol )
142 up = mod( myrow+nprow-1, nprow )
143 down = mod( myrow+1, nprow )
144 num = nprow*npcol
145
146
147
148 modkm1 = mod( m+1, hbl )
149 IF( modkm1.EQ.0 ) THEN
150 IF( ( myrow.EQ.ii ) .AND. ( right.EQ.jj ) .AND.
151 $ ( npcol.GT.1 ) ) THEN
152 CALL infog2l( m+2, m+1, desca, nprow, npcol, myrow, mycol,
153 $ irow, icol, rsrc, jsrc )
154 buf( 1 ) = a( ( icol-1 )*lda+irow )
155 CALL dgesd2d( contxt, 1, 1, buf, 1, ii, jj )
156 END IF
157 IF( ( down.EQ.ii ) .AND. ( right.EQ.jj ) .AND. ( num.GT.1 ) )
158 $ THEN
159 CALL infog2l( m, m, desca, nprow, npcol, myrow, mycol, irow,
160 $ icol, rsrc, jsrc )
161 buf( 1 ) = a( ( icol-1 )*lda+irow )
162 buf( 2 ) = a( ( icol-1 )*lda+irow+1 )
163 buf( 3 ) = a( icol*lda+irow )
164 buf( 4 ) = a( icol*lda+irow+1 )
165 CALL dgesd2d( contxt, 4, 1, buf, 4, ii, jj )
166 END IF
167 IF( ( myrow.EQ.ii ) .AND. ( mycol.EQ.jj ) ) THEN
168 CALL infog2l( m+2, m+2, desca, nprow, npcol, myrow, mycol,
169 $ irow, icol, rsrc, jsrc )
170 IF( npcol.GT.1 ) THEN
171 CALL dgerv2d( contxt, 1, 1, v3, 1, myrow, left )
172 ELSE
173 v3( 1 ) = a( ( icol-2 )*lda+irow )
174 END IF
175 IF( num.GT.1 ) THEN
176 CALL dgerv2d( contxt, 4, 1, buf, 4, up, left )
177 h11( 1 ) = buf( 1 )
178 h21( 1 ) = buf( 2 )
179 h12( 1 ) = buf( 3 )
180 h22 = buf( 4 )
181 ELSE
182 h11( 1 ) = a( ( icol-3 )*lda+irow-2 )
183 h21( 1 ) = a( ( icol-3 )*lda+irow-1 )
184 h12( 1 ) = a( ( icol-2 )*lda+irow-2 )
185 h22 = a( ( icol-2 )*lda+irow-1 )
186 END IF
187 END IF
188 END IF
189 IF( modkm1.EQ.1 ) THEN
190 IF( ( down.EQ.ii ) .AND. ( right.EQ.jj ) .AND. ( num.GT.1 ) )
191 $ THEN
192 CALL infog2l( m, m, desca, nprow, npcol, myrow, mycol, irow,
193 $ icol, rsrc, jsrc )
194 CALL dgesd2d( contxt, 1, 1, a( ( icol-1 )*lda+irow ), 1, ii,
195 $ jj )
196 END IF
197 IF( ( down.EQ.ii ) .AND. ( mycol.EQ.jj ) .AND. ( nprow.GT.1 ) )
198 $ THEN
199 CALL infog2l( m, m+1, desca, nprow, npcol, myrow, mycol,
200 $ irow, icol, rsrc, jsrc )
201 CALL dgesd2d( contxt, 1, 1, a( ( icol-1 )*lda+irow ), 1, ii,
202 $ jj )
203 END IF
204 IF( ( myrow.EQ.ii ) .AND. ( right.EQ.jj ) .AND.
205 $ ( npcol.GT.1 ) ) THEN
206 CALL infog2l( m+1, m, desca, nprow, npcol, myrow, mycol,
207 $ irow, icol, rsrc, jsrc )
208 CALL dgesd2d( contxt, 1, 1, a( ( icol-1 )*lda+irow ), 1, ii,
209 $ jj )
210 END IF
211 IF( ( myrow.EQ.ii ) .AND. ( mycol.EQ.jj ) ) THEN
212 CALL infog2l( m+2, m+2, desca, nprow, npcol, myrow, mycol,
213 $ irow, icol, rsrc, jsrc )
214 IF( num.GT.1 ) THEN
215 CALL dgerv2d( contxt, 1, 1, h11, 1, up, left )
216 ELSE
217 h11( 1 ) = a( ( icol-3 )*lda+irow-2 )
218 END IF
219 IF( nprow.GT.1 ) THEN
220 CALL dgerv2d( contxt, 1, 1, h12, 1, up, mycol )
221 ELSE
222 h12( 1 ) = a( ( icol-2 )*lda+irow-2 )
223 END IF
224 IF( npcol.GT.1 ) THEN
225 CALL dgerv2d( contxt, 1, 1, h21, 1, myrow, left )
226 ELSE
227 h21( 1 ) = a( ( icol-3 )*lda+irow-1 )
228 END IF
229 h22 = a( ( icol-2 )*lda+irow-1 )
230 v3( 1 ) = a( ( icol-2 )*lda+irow )
231 END IF
232 END IF
233 IF( ( myrow.NE.ii ) .OR. ( mycol.NE.jj ) )
234 $ RETURN
235
236 IF( modkm1.GT.1 ) THEN
237 CALL infog2l( m+2, m+2, desca, nprow, npcol, myrow, mycol,
238 $ irow, icol, rsrc, jsrc )
239 h11( 1 ) = a( ( icol-3 )*lda+irow-2 )
240 h21( 1 ) = a( ( icol-3 )*lda+irow-1 )
241 h12( 1 ) = a( ( icol-2 )*lda+irow-2 )
242 h22 = a( ( icol-2 )*lda+irow-1 )
243 v3( 1 ) = a( ( icol-2 )*lda+irow )
244 END IF
245
246 h44s = h44 - h11( 1 )
247 h33s = h33 - h11( 1 )
248 v1 = ( h33s*h44s-h43h34 ) / h21( 1 ) + h12( 1 )
249 v2 = h22 - h11( 1 ) - h33s - h44s
250 s = abs( v1 ) + abs( v2 ) + abs( v3( 1 ) )
251 v1 = v1 / s
252 v2 = v2 / s
253 v3( 1 ) = v3( 1 ) / s
254 v( 1 ) = v1
255 v( 2 ) = v2
256 v( 3 ) = v3( 1 )
257
258 RETURN
259
260
261
subroutine infog2l(grindx, gcindx, desc, nprow, npcol, myrow, mycol, lrindx, lcindx, rsrc, csrc)