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
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242 char DiagA, DirB, OpC, OpR, SideOp, TopC, TopR, TranOp, UploA,
243 Var, ctop, ctopsave, rtop, rtopsave;
244 Int Ai, Aj, Bi, Bj, ChooseAB, ForceTop, ctxt, info, itmp, lside,
245 mycol, myrow, nb, notran, nounit, npcol, nprow, upper;
246 double ABestL, ABestR, Best, tmp1, tmp2, tmp3, tmp4;
248
249
250
252
253
254
255
262#ifndef NO_ARGCHK
263
264
265
267 if( !( info = ( ( nprow == -1 ) ? -( 1101 +
CTXT_ ) : 0 ) ) )
268 {
269 if( ( !lside ) && ( SideOp !=
CRIGHT ) )
270 {
271 PB_Cwarn( ctxt, __LINE__,
"PSTRSM",
"Illegal SIDE = %c\n", SideOp );
272 info = -1;
273 }
274 else if( ( !upper ) && ( UploA !=
CLOWER ) )
275 {
276 PB_Cwarn( ctxt, __LINE__,
"PSTRSM",
"Illegal UPLO = %c\n", UploA );
277 info = -2;
278 }
279 else if( ( !notran ) && ( TranOp !=
CTRAN ) && ( TranOp !=
CCOTRAN ) )
280 {
281 PB_Cwarn( ctxt, __LINE__,
"PSTRSM",
"Illegal TRANS = %c\n", TranOp );
282 info = -3;
283 }
284 else if( ( !nounit ) && ( DiagA !=
CUNIT ) )
285 {
286 PB_Cwarn( ctxt, __LINE__,
"PSTRSM",
"Illegal DIAG = %c\n", DiagA );
287 info = -4;
288 }
289 if( lside )
290 PB_Cchkmat( ctxt,
"PSTRSM",
"A", *M, 5, *M, 5, Ai, Aj, Ad, 11,
291 &info );
292 else
293 PB_Cchkmat( ctxt,
"PSTRSM",
"A", *N, 6, *N, 6, Ai, Aj, Ad, 11,
294 &info );
295 PB_Cchkmat( ctxt,
"PSTRSM",
"B", *M, 5, *N, 6, Bi, Bj, Bd, 15,
296 &info );
297 }
298 if( info ) {
PB_Cabort( ctxt,
"PSTRSM", info );
return; }
299#endif
300
301
302
303 if( *M == 0 || *N == 0 ) return;
304
305
306
308
309
310
312 {
314 ((char *) B), Bi, Bj, Bd );
315 return;
316 }
317
318
319
320#ifdef NO_ARGCHK
322#endif
323
324
325
326
328
329
330
331
332
333
334 if( lside )
335 {
336 if( notran )
337 {
339 ABestR = (double)(*M) *
340 ( ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp1 /
TWO ) +
341 ( ( ( Bd[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp4 ) );
343 Best = (double)(*N) *
344 ( (double)(
CEIL( *M, itmp )) * (
double)(itmp) *
345 ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO :
ONE ) +
346 ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO :
ONE ) );
347 ChooseAB = ( ABestR <= ( 2.0 * Best ) );
348 }
349 else
350 {
353 ABestL = (double)(*M) *
354 ( ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp1 /
TWO ) +
356 ( ( ( Bd[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp4 ) );
357 ABestR = (double)(*M) *
358 ( ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp1 /
TWO ) +
359 ( ( ( Bd[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp4 ) +
360 MAX( tmp2, tmp1 ) /
TWO );
363 Best = (double)(*N) *
364 ( (double)(
CEIL( *M, itmp )) * (
double)(itmp) *
365 ( ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO :
ONE ) +
366 ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO :
ONE ) ) +
368 ChooseAB = ( ( ABestL <= ( 2.0 * Best ) ) ||
369 ( ABestR <= ( 2.0 * Best ) ) );
370 }
371 }
372 else
373 {
374 if( notran )
375 {
377 ABestR = (double)(*N) *
378 ( ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp2 /
TWO ) +
379 ( ( ( Bd[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp3 ) );
381 Best = (double)(*M) *
382 ( (double)(
CEIL( *N, itmp )) * (
double)(itmp) *
383 ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO :
ONE ) +
384 ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO :
ONE ) );
385 ChooseAB = ( ABestR <= ( 2.0 * Best ) );
386 }
387 else
388 {
391 ABestL = (double)(*N) *
392 ( ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp2 /
TWO ) +
394 ( ( ( Bd[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp3 ) );
395 ABestR = (double)(*N) *
396 ( ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO : tmp2 /
TWO ) +
397 ( ( ( Bd[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO : tmp3 ) +
398 MAX( tmp2, tmp1 ) /
TWO );
401 Best = (double)(*M) *
402 ( (double)(
CEIL( *N, itmp )) * (
double)(itmp) *
403 ( ( ( ( Ad[
RSRC_] == -1 ) || ( nprow == 1 ) ) ?
ZERO :
ONE ) +
404 ( ( ( Ad[
CSRC_] == -1 ) || ( npcol == 1 ) ) ?
ZERO :
ONE ) ) +
406 ChooseAB = ( ( ABestL <= ( 2.0 * Best ) ) ||
407 ( ABestR <= ( 2.0 * Best ) ) );
408 }
409 }
410
411
412
413
414
415
416
417
419
420
421 if( ChooseAB )
422 {
423
424
425
426
427
428 ForceTop = ( ( *M > nb ) && ( *N > nb ) );
429
430 if( ForceTop )
431 {
432 if( lside )
433 {
434 if( notran )
435 {
439 }
440 else
441 {
442 if( ABestL <= ABestR )
443 {
447 }
448 else
449 {
453 }
454 }
455 }
456 else
457 {
458 if( notran )
459 {
463 }
464 else
465 {
466 if( ABestL <= ABestR )
467 {
471 }
472 else
473 {
477 }
478 }
479 }
480
483
484 if( ( rtopsave = rtop ) != TopR )
486 if( ( ctopsave = ctop ) != TopC )
488
489
490
491
496 }
497
499 &DiagA, *M, *N, ((char *)ALPHA), ((char *)A), Ai, Aj, Ad,
500 ((char *)B), Bi, Bj, Bd );
501
502
503
504 if( ForceTop )
505 {
506 rtopsave = *
PB_Ctop( &ctxt, &OpR,
ROW, &rtopsave );
508 }
509 }
510 else
511 {
512
513
514
515 if( ( lside && notran ) || ( !lside && !notran ) )
516 {
520
521
522
523
525 }
526 else
527 {
531
532
533
534
536 }
537
540
541 if( ( rtopsave = rtop ) != TopR )
543 if( ( ctopsave = ctop ) != TopC )
545
548
550 &DiagA, *M, *N, ((char *)ALPHA), ((char *)A), Ai, Aj, Ad,
551 ((char *)B), Bi, Bj, Bd );
552
553
554
555 rtopsave = *
PB_Ctop( &ctxt, &OpR,
ROW, &rtopsave );
557 }
558
559
560
561}