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__,
"PCTRSM",
"Illegal SIDE = %c\n", SideOp );
272 info = -1;
273 }
274 else if( ( !upper ) && ( UploA !=
CLOWER ) )
275 {
276 PB_Cwarn( ctxt, __LINE__,
"PCTRSM",
"Illegal UPLO = %c\n", UploA );
277 info = -2;
278 }
279 else if( ( !notran ) && ( TranOp !=
CTRAN ) && ( TranOp !=
CCOTRAN ) )
280 {
281 PB_Cwarn( ctxt, __LINE__,
"PCTRSM",
"Illegal TRANS = %c\n", TranOp );
282 info = -3;
283 }
284 else if( ( !nounit ) && ( DiagA !=
CUNIT ) )
285 {
286 PB_Cwarn( ctxt, __LINE__,
"PCTRSM",
"Illegal DIAG = %c\n", DiagA );
287 info = -4;
288 }
289 if( lside )
290 PB_Cchkmat( ctxt,
"PCTRSM",
"A", *M, 5, *M, 5, Ai, Aj, Ad, 11,
291 &info );
292 else
293 PB_Cchkmat( ctxt,
"PCTRSM",
"A", *N, 6, *N, 6, Ai, Aj, Ad, 11,
294 &info );
295 PB_Cchkmat( ctxt,
"PCTRSM",
"B", *M, 5, *N, 6, Bi, Bj, Bd, 15,
296 &info );
297 }
298 if( info ) {
PB_Cabort( ctxt,
"PCTRSM", 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 if( ChooseAB )
421 {
422
423
424
425
426
427 ForceTop = ( ( *M > nb ) && ( *N > nb ) );
428
429 if( ForceTop )
430 {
431 if( lside )
432 {
433 if( notran )
434 {
438 }
439 else
440 {
441 if( ABestL <= ABestR )
442 {
446 }
447 else
448 {
452 }
453 }
454 }
455 else
456 {
457 if( notran )
458 {
462 }
463 else
464 {
465 if( ABestL <= ABestR )
466 {
470 }
471 else
472 {
476 }
477 }
478 }
479
482
483 if( ( rtopsave = rtop ) != TopR )
485 if( ( ctopsave = ctop ) != TopC )
487
488
489
490
495 }
496
497 PB_CptrsmAB( type, &Var, &SideOp, &UploA, &TranOp, &DiagA, *M, *N,
498 ((char *)ALPHA), ((char *)A), Ai, Aj, Ad, ((char *)B),
499 Bi, Bj, Bd );
500
501
502
503 if( ForceTop )
504 {
505 rtopsave = *
PB_Ctop( &ctxt, &OpR,
ROW, &rtopsave );
507 }
508 }
509 else
510 {
511
512
513
514 if( ( lside && notran ) || ( !lside && !notran ) )
515 {
519
520
521
522
524 }
525 else
526 {
530
531
532
533
535 }
536
539
540 if( ( rtopsave = rtop ) != TopR )
542 if( ( ctopsave = ctop ) != TopC )
544
547
548 PB_CptrsmB( type, &DirB, &SideOp, &UploA, &TranOp, &DiagA, *M, *N,
549 ((char *)ALPHA), ((char *)A), Ai, Aj, Ad, ((char *)B),
550 Bi, Bj, Bd );
551
552
553
554 rtopsave = *
PB_Ctop( &ctxt, &OpR,
ROW, &rtopsave );
556 }
557
558
559
560}