1 SUBROUTINE pzlarfc( SIDE, M, N, V, IV, JV, DESCV, INCV, TAU,
2 $ C, IC, JC, DESCC, WORK )
11 INTEGER IC, INCV, IV, JC, JV, , N
14 INTEGER DESCC( * ), ( * )
15 COMPLEX*16 C( * ), TAU( * ), V( * ), WORK( * )
229 INTEGER BLOCK_CYCLIC_2D, CSRC_, CTXT_, DLEN_, DTYPE_,
230 $ lld_, mb_, m_, nb_, n_, rsrc_
231 parameter( block_cyclic_2d = 1, dlen_ = 9, dtype_ = 1,
232 $ ctxt_ = 2, m_ = 3, n_ = 4, mb_ = 5, nb_ = 6,
233 $ rsrc_ = 7, csrc_ = 8, lld_ = 9 )
235 parameter( one = ( 1.0d+0, 0.0d+0 ),
236 $ zero = ( 0.0d+0, 0.0d+0 ) )
240 CHARACTER COLBTOP, ROWBTOP
241 INTEGER ICCOL, ICOFF, ICROW, ICTXT, IIC, IIV, IOFFC,
242 $ ioffv, ipw, iroff, ivcol, ivrow, jjc, jjv, ldc,
243 $ ldv, mycol, myrow, mp, ncc, ncv, npcol, nprow,
245 COMPLEX*16 TAULOC( 1 )
250 $
zgerc, zgerv2d, zgesd2d, zgsum2d,
256 EXTERNAL lsame, numroc
265 IF( m.LE.0 .OR. n.LE.0 )
270 ictxt = descc( ctxt_ )
275 CALL infog2l( ic, jc, descc, nprow, npcol, myrow, mycol, iic, jjc,
277 CALL infog2l( iv, jv, descv, nprow, npcol, myrow, mycol, iiv, jjv,
279 ncc = numroc( descc( n_ ), descc( nb_ ), mycol, descc
281 ncv = numroc( descv( n_ ), descv( nb_ ), mycol, descv( csrc_ ),
285 iic =
min( iic, ldc )
286 iiv =
min( iiv, ldv )
287 jjc =
min( jjc, ncc )
288 jjv =
min( jjv, ncv )
289 ioffc = iic+(jjc-1)*ldc
290 ioffv = iiv+(jjv-1)*ldv
292 iroff = mod( ic-1, descc( mb_ ) )
293 icoff = mod( jc-1, descc( nb_ ) )
294 mp = numroc( m+iroff, descc( mb_ ), myrow, icrow, nprow )
295 nq = numroc( n+icoff, descc( nb_ ), mycol, iccol, npcol )
303 crblck = ( m.LE.(descc( mb_ )-iroff) )
307 ccblck = ( n.LE.(descc( nb_ )-icoff) )
309 IF( lsame( side,
'L' ) )
THEN
321 IF( descv( m_ ).EQ.incv )
THEN
326 CALL pbztrnv( ictxt,
'Rowwise',
'Transpose', m,
327 $ descv( nb_ ), iroff, v( ioffv ), ldv, zero,
328 $ work, 1, ivrow, ivcol, icrow, iccol,
333 IF( mycol.EQ.iccol )
THEN
335 IF( myrow.EQ.ivrow )
THEN
337 CALL zgebs2d( ictxt,
'Columnwise',
' ', 1, 1,
339 tauloc( 1 ) = dconjg( tau( iiv ) )
343 CALL zgebr2d( ictxt,
'Columnwise',
' ', 1, 1,
344 $ tauloc, 1, ivrow, mycol )
349 IF( tauloc( 1 ).NE.zero )
THEN
354 CALL zgemv(
'Conjugate transpose', mp, nq, one,
355 $ c( ioffc ), ldc, work, 1, zero,
358 CALL zlaset(
'All', nq, 1, zero, zero,
359 $ work( ipw ),
max( 1, nq ) )
361 CALL zgsum2d( ictxt,
'Columnwise',
' ', nq, 1,
362 $ work( ipw ),
max( 1, nq ), rdest,
367 CALL zgerc( mp, nq, -tauloc( 1 ), work, 1,
368 $ work( ipw ), 1, c( ioffc ), ldc )
377 IF( ivcol.EQ.iccol )
THEN
381 IF( mycol.EQ.iccol )
THEN
383 tauloc( 1 ) = dconjg( tau( jjv ) )
385 IF( tauloc( 1 ).NE.zero )
THEN
390 CALL zgemv(
'Conjugate transpose', mp, nq,
391 $ one, c( ioffc ), ldc
394 CALL zlaset(
'All', nq, 1, zero, zero,
395 $ work,
max( 1, nq ) )
397 CALL zgsum2d( ictxt,
'Columnwise',
' ', nq, 1,
398 $ work,
max( 1, nq ), rdest, mycol )
402 CALL zgerc( mp, nq, -tauloc( 1 ), v( ioffv ), 1,
403 $ work, 1, c( ioffc ), ldc )
412 IF( mycol.EQ.ivcol )
THEN
415 CALL zcopy( mp, v( ioffv ), 1, work, 1 )
416 work( ipw ) = tau( jjv )
417 CALL zgesd2d( ictxt, ipw, 1, work, ipw, myrow,
420 ELSE IF( mycol.EQ.iccol )
THEN
423 CALL zgerv2d( ictxt, ipw, 1, work, ipw, myrow,
425 tauloc( 1 ) = dconjg( work( ipw ) )
427 IF( tauloc( 1 ).NE.zero )
THEN
432 CALL zgemv(
'Conjugate transpose', mp, nq,
433 $ one, c( ioffc ), ldc, work, 1,
434 $ zero, work( ipw ), 1 )
436 CALL zlaset(
'All', nq, 1, zero, zero,
437 $ work( ipw ),
max( 1, nq ) )
439 CALL zgsum2d( ictxt,
'Columnwise',
' ', nq, 1,
440 $ work( ipw ),
max( 1, nq ), rdest,
445 CALL zgerc( mp, nq, -tauloc( 1 ), work, 1,
446 $ work( ipw ), 1, c( ioffc ), ldc )
459 IF( descv( m_ ).EQ.incv )
THEN
464 CALL pbztrnv( ictxt,
'Rowwise',
'Transpose', m,
465 $ descv( nb_ ), iroff, v( ioffv ), ldv, zero,
466 $ work, 1, ivrow, ivcol, icrow, -1,
471 IF( myrow.EQ.ivrow )
THEN
473 CALL zgebs2d( ictxt,
'Columnwise',
' ', 1, 1,
475 tauloc( 1 ) = dconjg( tau( iiv ) )
479 CALL zgebr2d( ictxt,
'Columnwise',
' ', 1, 1, tauloc,
481 tauloc( 1 ) = dconjg( tauloc( 1 ) )
485 IF( tauloc( 1 ).NE.zero )
THEN
490 CALL zgemv(
'Conjugate transpose', mp, nq, one,
491 $ c( ioffc ), ldc, work, 1, zero,
494 CALL zlaset(
'All', nq, 1, zero, zero,
495 $ work( ipw ),
max( 1, nq ) )
497 CALL zgsum2d( ictxt,
'Columnwise',
' ', nq, 1,
498 $ work( ipw ),
max( 1, nq ), rdest,
503 CALL zgerc( mp, nq, -tauloc( 1 ), work, 1,
504 $ work( ipw ), 1, c( ioffc ), ldc )
511 CALL pb_topget( ictxt,
'Broadcast',
'Rowwise', rowbtop
512 IF( mycol.EQ.ivcol )
THEN
515 CALL zcopy( mp, v( ioffv ), 1, work, 1 )
516 work(ipw) = tau( jjv )
517 CALL zgebs2d( ictxt,
'Rowwise', rowbtop, ipw, 1,
519 tauloc( 1 ) = dconjg( tau( jjv ) )
524 CALL zgebr2d( ictxt,
'Rowwise', rowbtop, ipw, 1, work,
525 $ ipw, myrow, ivcol )
526 tauloc( 1 ) = dconjg( work( ipw ) )
530 IF( tauloc( 1 ).NE.zero )
THEN
535 CALL zgemv(
'Conjugate transpose', mp, nq, one,
536 $ c( ioffc ), ldc, work, 1, zero,
539 CALL zlaset(
'All', nq, 1, zero, zero,
540 $ work( ipw ),
max( 1, nq ) )
542 CALL zgsum2d( ictxt,
'Columnwise',
' ', nq, 1,
543 $ work( ipw ),
max( 1, nq ), rdest,
548 CALL zgerc( mp, nq, -tauloc( 1 ), work, 1,
549 $ work( ipw ), 1, c( ioffc ), ldc )
568 IF( descv( m_ ).EQ.incv )
THEN
572 IF( ivrow.EQ.icrow )
THEN
576 IF( myrow.EQ.icrow )
THEN
578 tauloc( 1 ) = dconjg( tau( iiv ) )
580 IF( tauloc( 1 ).NE.zero )
THEN
585 CALL zgemv(
'No transpose', mp, nq, one,
586 $ c( ioffc ), ldc, v( ioffv ), ldv,
589 CALL zlaset(
'All', mp, 1, zero, zero,
590 $ work,
max( 1, mp ) )
592 CALL zgsum2d( ictxt, 'rowwise
', ' ', MP, 1,
593 $ WORK, MAX( 1, MP ), RDEST, ICCOL )
597 CALL ZGERC( MP, NQ, -TAULOC( 1 ), WORK, 1,
598 $ V( IOFFV ), LDV, C( IOFFC ), LDC )
607.EQ.
IF( MYROWIVROW ) THEN
610 CALL ZCOPY( NQ, V( IOFFV ), LDV, WORK, 1 )
611 WORK(IPW) = TAU( IIV )
612 CALL ZGESD2D( ICTXT, IPW, 1, WORK, IPW, ICROW,
615.EQ.
ELSE IF( MYROWICROW ) THEN
618 CALL ZGERV2D( ICTXT, IPW, 1, WORK, IPW, IVROW,
620 TAULOC( 1 ) = DCONJG( WORK( IPW ) )
622.NE.
IF( TAULOC( 1 )ZERO ) THEN
627 CALL ZGEMV( 'no transpose
', MP, NQ, ONE,
628 $ C( IOFFC ), LDC, WORK, 1, ZERO,
631 CALL ZLASET( 'all
', MP, 1, ZERO, ZERO,
632 $ WORK( IPW ), MAX( 1, MP ) )
634 CALL ZGSUM2D( ICTXT, 'rowwise
', '', MP, 1,
635 $ WORK( IPW ), MAX( 1, MP ), RDEST,
640 CALL ZGERC( MP, NQ, -TAULOC( 1 ), WORK( IPW ),
641 $ 1, WORK, 1, C( IOFFC ), LDC )
653 CALL PBZTRNV( ICTXT, 'columnwise
', 'transpose
', N,
654 $ DESCV( MB_ ), ICOFF, V( IOFFV ), 1, ZERO,
655 $ WORK, 1, IVROW, IVCOL, ICROW, ICCOL,
660.EQ.
IF( MYROWICROW ) THEN
662.EQ.
IF( MYCOLIVCOL ) THEN
664 CALL ZGEBS2D( ICTXT, 'rowwise
', ' ', 1, 1,
666 TAULOC( 1 ) = DCONJG( TAU( JJV ) )
670 CALL ZGEBR2D( ICTXT, 'rowwise
', ' ', 1, 1, TAULOC,
672 TAULOC( 1 ) = DCONJG( TAULOC( 1 ) )
676.NE.
IF( TAULOC( 1 )ZERO ) THEN
681 CALL ZGEMV( 'no transpose
', MP, NQ, ONE,
682 $ C( IOFFC ), LDC, WORK, 1, ZERO,
685 CALL ZLASET( 'all
', MP, 1, ZERO, ZERO,
686 $ WORK( IPW ), MAX( 1, MP ) )
688 CALL ZGSUM2D( ICTXT, 'rowwise
', ' ', MP, 1,
689 $ WORK( IPW ), MAX( 1, MP ), RDEST,
694 CALL ZGERC( MP, NQ, -TAULOC( 1 ), WORK( IPW ), 1,
695 $ WORK, 1, C( IOFFC ), LDC )
706.EQ.
IF( DESCV( M_ )INCV ) THEN
710 CALL PB_TOPGET( ICTXT, 'broadcast
', 'columnwise
',
712.EQ.
IF( MYROWIVROW ) THEN
715 CALL ZCOPY( NQ, V( IOFFV ), LDV, WORK, 1 )
716 WORK(IPW) = TAU( IIV )
717 CALL ZGEBS2D( ICTXT, 'columnwise
', COLBTOP, IPW, 1,
719 TAULOC( 1 ) = DCONJG( TAU( IIV ) )
724 CALL ZGEBR2D( ICTXT, 'columnwise
', COLBTOP, IPW, 1,
725 $ WORK, IPW, IVROW, MYCOL )
726 TAULOC( 1 ) = DCONJG( WORK( IPW ) )
730.NE.
IF( TAULOC( 1 )ZERO ) THEN
735 CALL ZGEMV( 'no transpose
', MP, NQ, ONE,
736 $ C( IOFFC ), LDC, WORK, 1, ZERO,
739 CALL ZLASET( 'all
', MP, 1, ZERO, ZERO,
740 $ WORK( IPW ), MAX( 1, MP ) )
742 CALL ZGSUM2D( ICTXT, 'rowwise
', ' ', MP, 1,
743 $ WORK( IPW ), MAX( 1, MP ), RDEST,
748 CALL ZGERC( MP, NQ, -TAULOC( 1 ), WORK( IPW ), 1,
749 $ WORK, 1, C( IOFFC ), LDC )
757 CALL PBZTRNV( ICTXT, 'columnwise
', 'transpose
', N,
758 $ DESCV( MB_ ), ICOFF, V( IOFFV ), 1, ZERO,
759 $ WORK, 1, IVROW, IVCOL, -1, ICCOL,
764.EQ.
IF( MYCOLIVCOL ) THEN
766 CALL ZGEBS2D( ICTXT, 'rowwise
', ' ', 1, 1, TAU( JJV ),
768 TAULOC( 1 ) = DCONJG( TAU( JJV ) )
772 CALL ZGEBR2D( ICTXT, 'rowwise
', ' ', 1, 1, TAULOC, 1,
774 TAULOC( 1 ) = DCONJG( TAULOC( 1 ) )
778.NE.
IF( TAULOC( 1 )ZERO ) THEN
783 CALL ZGEMV( 'no transpose
', MP, NQ, ONE,
784 $ C( IOFFC ), LDC, WORK, 1, ZERO,
787 CALL ZLASET( 'all
', MP, 1, ZERO, ZERO, WORK( IPW ),
790 CALL ZGSUM2D( ICTXT, 'rowwise
', ' ', MP, 1,
791 $ WORK( IPW ), MAX( 1, MP ), RDEST,
796 CALL ZGERC( MP, NQ, -TAULOC( 1 ), WORK( IPW ), 1,
797 $ WORK, 1, C( IOFFC ), LDC )