! ! Parallel Sparse BLAS version 3.5 ! (C) Copyright 2006, 2010, 2015, 2017 ! Salvatore Filippone Cranfield University ! Alfredo Buttari CNRS-IRIT, Toulouse ! ! Redistribution and use in source and binary forms, with or without ! modification, are permitted provided that the following conditions ! are met: ! 1. Redistributions of source code must retain the above copyright ! notice, this list of conditions and the following disclaimer. ! 2. Redistributions in binary form must reproduce the above copyright ! notice, this list of conditions, and the following disclaimer in the ! documentation and/or other materials provided with the distribution. ! 3. The name of the PSBLAS group or the names of its contributors may ! not be used to endorse or promote products derived from this ! software without specific written permission. ! ! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ! ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR ! PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PSBLAS GROUP OR ITS CONTRIBUTORS ! BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR ! CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF ! SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS ! INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN ! CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ! ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE ! POSSIBILITY OF SUCH DAMAGE. ! ! ! ! File: psi_sswaptran.F90 ! ! Subroutine: psi_sswaptranm ! Does the data exchange among processes. This is similar to Xswapdata, but ! the list is read "in reverse", i.e. indices that are normally SENT are used ! for the RECEIVE part and vice-versa. This is the basic data exchange operation ! for doing the product of a sparse matrix by a vector. ! Essentially this is doing a variable all-to-all data exchange ! (ALLTOALLV in MPI parlance), but ! it is capable of pruning empty exchanges, which are very likely in out ! application environment. All the variants have the same structure ! In all these subroutines X may be: I Integer ! S real(psb_spk_) ! D real(psb_dpk_) ! C complex(psb_spk_) ! Z complex(psb_dpk_) ! Basically the operation is as follows: on each process, we identify ! sections SND(Y) and RCV(Y); then we do a SEND(PACK(SND(Y))); ! then we receive, and we do an update with Y = UNPACK(RCV(Y)) + BETA * Y ! but only on the elements involved in the UNPACK operation. ! Thus: for halo data exchange, the receive section is confined in the ! halo indices, and BETA=0, whereas for overlap exchange the receive section ! is scattered in the owned indices, and BETA=1. ! ! Arguments: ! flag - integer Choose the algorithm for data exchange: ! this is chosen through bit fields. ! swap_mpi = iand(flag,psb_swap_mpi_) /= 0 ! swap_sync = iand(flag,psb_swap_sync_) /= 0 ! swap_send = iand(flag,psb_swap_send_) /= 0 ! swap_recv = iand(flag,psb_swap_recv_) /= 0 ! if (swap_mpi): use underlying MPI_ALLTOALLV. ! if (swap_sync): use PSB_SND and PSB_RCV in ! synchronized pairs ! if (swap_send .and. swap_recv): use mpi_irecv ! and mpi_send ! if (swap_send): use psb_snd (but need another ! call with swap_recv to complete) ! if (swap_recv): use psb_rcv (completing a ! previous call with swap_send) ! ! ! n - integer Number of columns in Y ! beta - X Choose overwrite or sum. ! y(:,:) - X The data area ! desc_a - type(psb_desc_type). The communication descriptor. ! work(:) - X Buffer space. If not sufficient, will do ! our own internal allocation. ! info - integer. return code. ! data - integer which list is to be used to exchange data ! default psb_comm_halo_ ! psb_comm_halo_ use halo_index ! psb_comm_ext_ use ext_index ! psb_comm_ovrl_ use ovrl_index ! psb_comm_mov_ use ovr_mst_idx ! ! subroutine psi_sswaptranm(flag,n,beta,y,desc_a,work,info,data) use psi_mod, psb_protect_name => psi_sswaptranm use psb_error_mod use psb_desc_mod use psb_penv_mod use psb_caf_mod #ifdef MPI_MOD use mpi #endif implicit none #ifdef MPI_H include 'mpif.h' #endif integer(psb_ipk_), intent(in) :: flag, n integer(psb_ipk_), intent(out) :: info real(psb_spk_) :: y(:,:), beta real(psb_spk_), target :: work(:) type(psb_desc_type),target :: desc_a integer(psb_ipk_), optional :: data ! locals integer(psb_ipk_) :: ictxt, np, me, icomm, idxs, idxr, err_act, totxch, data_ integer(psb_ipk_), pointer :: d_idx(:) class(psb_xch_idx_type), pointer :: d_xchg integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swap_tran' call psb_erractionsave(err_act) ictxt = desc_a%get_context() icomm = desc_a%get_mpic() call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif if (.not.psb_is_asb_desc(desc_a)) then info=psb_err_invalid_cd_state_ call psb_errpush(info,name) goto 9999 endif if(present(data)) then data_ = data else data_ = psb_comm_halo_ end if call desc_a%get_list(data_,d_idx,totxch,idxr,idxs,info) if (info == 0) call desc_a%get_list(data_,d_xchg,info) if (info /= psb_success_) then call psb_errpush(psb_err_internal_error_,name,a_err='psb_cd_get_list') goto 9999 end if if (if_caf2) then call psi_swaptran(ictxt,icomm,flag,n,beta,y,d_xchg,info) endif if (info /= psb_success_) goto 9999 call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptranm subroutine psi_sswaptran_xchg_m(iictxt,iicomm,flag,m,beta,y,xchg,info) use psi_mod, psb_protect_name => psi_sswaptran_xchg_m use psb_error_mod use psb_realloc_mod use psb_desc_mod use psb_penv_mod use psb_s_base_vect_mod use iso_fortran_env implicit none integer(psb_ipk_), intent(in) :: iictxt,iicomm,flag, m integer(psb_ipk_), intent(out) :: info real(psb_spk_) :: y(:,:) real(psb_spk_) :: beta class(psb_xch_idx_type), intent(inout) :: xchg ! locals integer(psb_mpik_) :: ictxt, icomm, np, me,& & proc_to_comm, p2ptag, iret integer(psb_ipk_) :: nesd, nerv,& & err_act, i, idx_pt, totsnd_, totrcv_,p1,p2,isz,rp1,rp2,& & snd_pt, rcv_pt, pnti, n, ip, img, nxch, myself integer :: count real(psb_spk_), allocatable, save :: buffer(:)[:], sndbuf(:) type(event_type), allocatable, save :: ufg(:)[:] type(event_type), allocatable, save :: clear[:] integer, save :: last_clear_count = 0 logical :: swap_mpi, swap_sync, swap_send, swap_recv,& & albf,do_send,do_recv integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swaptran_datam' call psb_erractionsave(err_act) ictxt = iictxt icomm = iicomm call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif n=1 swap_mpi = iand(flag,psb_swap_mpi_) /= 0 swap_sync = iand(flag,psb_swap_sync_) /= 0 swap_send = iand(flag,psb_swap_send_) /= 0 swap_recv = iand(flag,psb_swap_recv_) /= 0 do_send = swap_mpi .or. swap_sync .or. swap_send do_recv = swap_mpi .or. swap_sync .or. swap_recv if (.not.(do_send.and.do_recv)) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Unimplemented case in tran_xchg_vm') goto 9999 end if if (.not.allocated(ufg)) then !write(*,*) 'Allocating events',np allocate(ufg(np)[*],stat=info) if (info == 0) allocate(clear[*],stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray events allocation') goto 9999 end if else if (last_clear_count>0) & & event wait(clear,until_count=last_clear_count) end if if (psb_size(buffer) < xchg%max_buffer_size) then ! ! By construction, max_buffer_size was computed with a collective. ! if (allocated(buffer)) deallocate(buffer) if (allocated(sndbuf)) deallocate(sndbuf) !write(*,*) 'Allocating buffer',xchg%max_buffer_size allocate(buffer(xchg%max_buffer_size)[*],stat=info) if (info == 0) allocate(sndbuf(xchg%max_buffer_size),stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray buffer allocation') goto 9999 end if end if if (.false.) then nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 !write(0,*) myself,'Posting for ',img,' boundaries: ',p1,p2 call psi_gth(isz,m,xchg%loc_rcv_idx(p1:p2),y,buffer(p1:p2)) event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,'Remote boundaries: ',rp1,rp2 call psi_sct(isz,m,xchg%loc_snd_idx(p1:p2),buffer(rp1:rp2)[img],beta,y) event post(clear[img]) end do last_clear_count = nxch else nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 call psi_gth(isz,m,xchg%loc_rcv_idx(p1:p2),& & y,sndbuf(p1:p2)) buffer(rp1:rp2)[img] = sndbuf(p1:p2) end do ! ! Doing event post later should provide more opportunities for ! overlap ! do ip= 1, nxch img = xchg%prcs_xch(ip) + 1 event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,' boundaries: ',p1,p2 call psi_sct(isz,m,xchg%loc_snd_idx(p1:p2),& & buffer(p1:p2),beta, y) event post(clear[img]) end do last_clear_count = nxch endif call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptran_xchg_m ! ! ! Subroutine: psi_sswaptranv ! Does the data exchange among processes. This is similar to Xswapdata, but ! the list is read "in reverse", i.e. indices that are normally SENT are used ! for the RECEIVE part and vice-versa. This is the basic data exchange operation ! for doing the product of a sparse matrix by a vector. ! Essentially this is doing a variable all-to-all data exchange ! (ALLTOALLV in MPI parlance), but ! it is capable of pruning empty exchanges, which are very likely in out ! application environment. All the variants have the same structure ! In all these subroutines X may be: I Integer ! S real(psb_spk_) ! D real(psb_dpk_) ! C complex(psb_spk_) ! Z complex(psb_dpk_) ! Basically the operation is as follows: on each process, we identify ! sections SND(Y) and RCV(Y); then we do a SEND(PACK(SND(Y))); ! then we receive, and we do an update with Y = UNPACK(RCV(Y)) + BETA * Y ! but only on the elements involved in the UNPACK operation. ! Thus: for halo data exchange, the receive section is confined in the ! halo indices, and BETA=0, whereas for overlap exchange the receive section ! is scattered in the owned indices, and BETA=1. ! ! Arguments: ! flag - integer Choose the algorithm for data exchange: ! this is chosen through bit fields. ! swap_mpi = iand(flag,psb_swap_mpi_) /= 0 ! swap_sync = iand(flag,psb_swap_sync_) /= 0 ! swap_send = iand(flag,psb_swap_send_) /= 0 ! swap_recv = iand(flag,psb_swap_recv_) /= 0 ! if (swap_mpi): use underlying MPI_ALLTOALLV. ! if (swap_sync): use PSB_SND and PSB_RCV in ! synchronized pairs ! if (swap_send .and. swap_recv): use mpi_irecv ! and mpi_send ! if (swap_send): use psb_snd (but need another ! call with swap_recv to complete) ! if (swap_recv): use psb_rcv (completing a ! previous call with swap_send) ! ! ! n - integer Number of columns in Y ! beta - X Choose overwrite or sum. ! y(:) - X The data area ! desc_a - type(psb_desc_type). The communication descriptor. ! work(:) - X Buffer space. If not sufficient, will do ! our own internal allocation. ! info - integer. return code. ! data - integer which list is to be used to exchange data ! default psb_comm_halo_ ! psb_comm_halo_ use halo_index ! psb_comm_ext_ use ext_index ! psb_comm_ovrl_ use ovrl_index ! psb_comm_mov_ use ovr_mst_idx ! ! subroutine psi_sswaptranv(flag,beta,y,desc_a,work,info,data) use psi_mod, psb_protect_name => psi_sswaptranv use psb_error_mod use psb_desc_mod use psb_penv_mod use psb_caf_mod #ifdef MPI_MOD use mpi #endif implicit none #ifdef MPI_H include 'mpif.h' #endif integer(psb_ipk_), intent(in) :: flag integer(psb_ipk_), intent(out) :: info real(psb_spk_) :: y(:), beta real(psb_spk_), target :: work(:) type(psb_desc_type),target :: desc_a integer(psb_ipk_), optional :: data ! locals integer(psb_ipk_) :: ictxt, np, me, icomm, idxs, idxr, totxch, err_act, data_ integer(psb_ipk_), pointer :: d_idx(:) class(psb_xch_idx_type), pointer :: d_xchg integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swap_tranv' call psb_erractionsave(err_act) ictxt = desc_a%get_context() icomm = desc_a%get_mpic() call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif if (.not.psb_is_asb_desc(desc_a)) then info=psb_err_invalid_cd_state_ call psb_errpush(info,name) goto 9999 endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ end if call desc_a%get_list(data_,d_idx,totxch,idxr,idxs,info) if (info == 0) call desc_a%get_list(data_,d_xchg,info) if (info /= psb_success_) then call psb_errpush(psb_err_internal_error_,name,a_err='psb_cd_get_list') goto 9999 end if if (if_caf2) then call psi_swaptran(ictxt,icomm,flag,beta,y,d_xchg,info) endif if (info /= psb_success_) goto 9999 call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptranv subroutine psi_sswaptran_xchg_v(iictxt,iicomm,flag,beta,y,xchg,info) use psi_mod, psb_protect_name => psi_sswaptran_xchg_v use psb_error_mod use psb_realloc_mod use psb_desc_mod use psb_penv_mod use psb_s_base_vect_mod use iso_fortran_env implicit none integer(psb_ipk_), intent(in) :: iictxt,iicomm,flag integer(psb_ipk_), intent(out) :: info real(psb_spk_) :: y(:) real(psb_spk_) :: beta class(psb_xch_idx_type), intent(inout) :: xchg ! locals integer(psb_mpik_) :: ictxt, icomm, np, me,& & proc_to_comm, p2ptag, iret integer(psb_ipk_) :: nesd, nerv,& & err_act, i, idx_pt, totsnd_, totrcv_,p1,p2,isz,rp1,rp2,& & snd_pt, rcv_pt, pnti, n, ip, img, nxch, myself integer :: count real(psb_spk_), allocatable, save :: buffer(:)[:], sndbuf(:) type(event_type), allocatable, save :: ufg(:)[:] type(event_type), allocatable, save :: clear[:] integer, save :: last_clear_count = 0 logical :: swap_mpi, swap_sync, swap_send, swap_recv,& & albf,do_send,do_recv integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swap_datav' call psb_erractionsave(err_act) ictxt = iictxt icomm = iicomm call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif n=1 swap_mpi = iand(flag,psb_swap_mpi_) /= 0 swap_sync = iand(flag,psb_swap_sync_) /= 0 swap_send = iand(flag,psb_swap_send_) /= 0 swap_recv = iand(flag,psb_swap_recv_) /= 0 do_send = swap_mpi .or. swap_sync .or. swap_send do_recv = swap_mpi .or. swap_sync .or. swap_recv if (.not.(do_send.and.do_recv)) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Unimplemented case in xchg_vect') goto 9999 end if if (.not.allocated(ufg)) then !write(*,*) 'Allocating events',np allocate(ufg(np)[*],stat=info) if (info == 0) allocate(clear[*],stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray events allocation') goto 9999 end if else if (last_clear_count>0) & & event wait(clear,until_count=last_clear_count) end if if (psb_size(buffer) < xchg%max_buffer_size) then ! ! By construction, max_buffer_size was computed with a collective. ! if (allocated(buffer)) deallocate(buffer) if (allocated(sndbuf)) deallocate(sndbuf) !write(*,*) 'Allocating buffer',xchg%max_buffer_size allocate(buffer(xchg%max_buffer_size)[*],stat=info) if (info == 0) allocate(sndbuf(xchg%max_buffer_size),stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray buffer allocation') goto 9999 end if end if if (.false.) then nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 !write(0,*) myself,'Posting for ',img,' boundaries: ',p1,p2 call psi_gth(isz,xchg%loc_rcv_idx(p1:p2),y,buffer(p1:p2)) event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,'Remote boundaries: ',rp1,rp2 call psi_sct(isz,xchg%loc_snd_idx(p1:p2),buffer(rp1:rp2)[img],beta,y) event post(clear[img]) end do last_clear_count = nxch else nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 call psi_gth(isz,xchg%loc_rcv_idx(p1:p2),& & y,sndbuf(p1:p2)) buffer(rp1:rp2)[img] = sndbuf(p1:p2) end do ! ! Doing event post later should provide more opportunities for ! overlap ! do ip= 1, nxch img = xchg%prcs_xch(ip) + 1 event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,' boundaries: ',p1,p2 call psi_sct(isz,xchg%loc_snd_idx(p1:p2),& & buffer(p1:p2),beta, y) event post(clear[img]) end do last_clear_count = nxch endif call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptran_xchg_v ! ! ! Subroutine: psi_sswaptran_vect ! Data exchange among processes. ! ! Takes care of Y an exanspulated vector. ! ! subroutine psi_sswaptran_vect(flag,beta,y,desc_a,work,info,data) use psi_mod, psb_protect_name => psi_sswaptran_vect use psb_s_base_vect_mod use psb_error_mod use psb_desc_mod use psb_penv_mod use psb_caf_mod #ifdef MPI_MOD use mpi #endif implicit none #ifdef MPI_H include 'mpif.h' #endif integer(psb_ipk_), intent(in) :: flag integer(psb_ipk_), intent(out) :: info class(psb_s_base_vect_type) :: y real(psb_spk_) :: beta real(psb_spk_), target :: work(:) type(psb_desc_type),target :: desc_a integer(psb_ipk_), optional :: data ! locals integer(psb_ipk_) :: ictxt, np, me, icomm, idxs, idxr, totxch, err_act, data_ class(psb_i_base_vect_type), pointer :: d_vidx class(psb_xch_idx_type), pointer :: d_xchg integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swap_tranv' call psb_erractionsave(err_act) ictxt = desc_a%get_context() icomm = desc_a%get_mpic() call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif if (.not.psb_is_asb_desc(desc_a)) then info=psb_err_invalid_cd_state_ call psb_errpush(info,name) goto 9999 endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ end if call desc_a%get_list(data_,d_vidx,totxch,idxr,idxs,info) if (info == 0) call desc_a%get_list(data_,d_xchg,info) if (info /= psb_success_) then call psb_errpush(psb_err_internal_error_,name,a_err='psb_cd_get_list') goto 9999 end if if (if_caf2) then call psi_swaptran(ictxt,icomm,flag,beta,y,d_xchg,info) endif if (info /= psb_success_) goto 9999 call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptran_vect subroutine psi_sswaptran_xchg_vect(iictxt,iicomm,flag,beta,y,xchg,info) use psi_mod, psb_protect_name => psi_sswaptran_xchg_vect use psb_error_mod use psb_realloc_mod use psb_desc_mod use psb_penv_mod use psb_s_base_vect_mod use iso_fortran_env implicit none integer(psb_ipk_), intent(in) :: iictxt,iicomm,flag integer(psb_ipk_), intent(out) :: info class(psb_s_base_vect_type) :: y real(psb_spk_) :: beta class(psb_xch_idx_type), intent(inout) :: xchg ! locals integer(psb_mpik_) :: ictxt, icomm, np, me,& & proc_to_comm, p2ptag, iret integer(psb_ipk_) :: nesd, nerv,& & err_act, i, idx_pt, totsnd_, totrcv_,p1,p2,isz,rp1,rp2,& & snd_pt, rcv_pt, pnti, n, ip, img, nxch, myself integer :: count real(psb_spk_), allocatable, save :: buffer(:)[:], sndbuf(:) type(event_type), allocatable, save :: ufg(:)[:] type(event_type), allocatable, save :: clear[:] integer, save :: last_clear_count = 0 logical :: swap_mpi, swap_sync, swap_send, swap_recv,& & albf,do_send,do_recv integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_tran_xchg_vect' call psb_erractionsave(err_act) ictxt = iictxt icomm = iicomm call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif if (np /= num_images()) then write(*,*) 'Something is wrong MPI vs CAF ', np, num_images() info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Num_images /= np') goto 9999 end if n=1 swap_mpi = iand(flag,psb_swap_mpi_) /= 0 swap_sync = iand(flag,psb_swap_sync_) /= 0 swap_send = iand(flag,psb_swap_send_) /= 0 swap_recv = iand(flag,psb_swap_recv_) /= 0 do_send = swap_mpi .or. swap_sync .or. swap_send do_recv = swap_mpi .or. swap_sync .or. swap_recv if (.not.(do_send.and.do_recv)) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Unimplemented case in xchg_vect') goto 9999 end if if (.not.allocated(ufg)) then !write(*,*) 'Allocating events',np allocate(ufg(np)[*],stat=info) if (info == 0) allocate(clear[*],stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray events allocation') goto 9999 end if else if (last_clear_count>0) & & event wait(clear,until_count=last_clear_count) end if if (psb_size(buffer) < xchg%max_buffer_size) then ! ! By construction, max_buffer_size was computed with a collective. ! if (allocated(buffer)) deallocate(buffer) !write(*,*) 'Allocating buffer',xchg%max_buffer_size allocate(buffer(xchg%max_buffer_size)[*],stat=info) if (allocated(sndbuf)) deallocate(sndbuf) if (info == 0) allocate(sndbuf(xchg%max_buffer_size),stat=info) if (info /= 0) then info = psb_err_internal_error_ call psb_errpush(info,name,a_err='Coarray buffer allocation') goto 9999 end if end if if (.false.) then !sync all nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 !write(0,*) myself,'Posting for ',img,' boundaries: ',p1,p2 call y%gth(isz,xchg%loc_rcv_idx(p1:p2),buffer(p1:p2)) event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,'Remote boundaries: ',rp1,rp2 call y%sct(isz,xchg%loc_snd_idx(p1:p2),buffer(rp1:rp2)[img],beta) event post(clear[img]) end do last_clear_count = nxch else nxch = size(xchg%prcs_xch) myself = this_image() do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_rcv_bnd(ip) p2 = xchg%loc_rcv_bnd(ip+1)-1 rp1 = xchg%rmt_snd_bnd(ip,1) rp2 = xchg%rmt_snd_bnd(ip,2) isz = p2-p1+1 !write(0,*) myself,'Posting for ',img,' boundaries: ',rp1,rp2 if (.false.) then call y%gth(isz,xchg%loc_rcv_idx(p1:p2),buffer(rp1:rp2)[img]) else call y%gth(isz,xchg%loc_rcv_idx(p1:p2),sndbuf(p1:p2)) buffer(rp1:rp2)[img] = sndbuf(p1:p2) end if end do ! ! Doing event post later should provide more opportunities for ! overlap ! do ip= 1, nxch img = xchg%prcs_xch(ip) + 1 event post(ufg(myself)[img]) end do do ip = 1, nxch img = xchg%prcs_xch(ip) + 1 event wait(ufg(img)) img = xchg%prcs_xch(ip) + 1 p1 = xchg%loc_snd_bnd(ip) p2 = xchg%loc_snd_bnd(ip+1)-1 isz = p2-p1+1 rp1 = xchg%rmt_rcv_bnd(ip,1) rp2 = xchg%rmt_rcv_bnd(ip,2) !write(0,*) myself,'Getting from ',img,' boundaries: ',p1,p2 call y%sct(isz,xchg%loc_snd_idx(p1:p2),buffer(p1:p2),beta) event post(clear[img]) end do last_clear_count = nxch end if call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptran_xchg_vect ! ! ! ! ! Subroutine: psi_sswaptran_vect ! Data exchange among processes. ! ! Takes care of Y an encaspulated vector. ! ! subroutine psi_sswaptran_multivect(flag,beta,y,desc_a,work,info,data) use psi_mod, psb_protect_name => psi_sswaptran_multivect use psb_s_base_vect_mod use psb_error_mod use psb_desc_mod use psb_penv_mod #ifdef MPI_MOD use mpi #endif implicit none #ifdef MPI_H include 'mpif.h' #endif integer(psb_ipk_), intent(in) :: flag integer(psb_ipk_), intent(out) :: info class(psb_s_base_multivect_type) :: y real(psb_spk_) :: beta real(psb_spk_), target :: work(:) type(psb_desc_type),target :: desc_a integer(psb_ipk_), optional :: data ! locals integer(psb_ipk_) :: ictxt, np, me, icomm, idxs, idxr, totxch, err_act, data_ class(psb_i_base_vect_type), pointer :: d_vidx integer(psb_ipk_) :: ierr(5) character(len=20) :: name info=psb_success_ name='psi_swap_tranv' call psb_erractionsave(err_act) ictxt = desc_a%get_context() icomm = desc_a%get_mpic() call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif if (.not.psb_is_asb_desc(desc_a)) then info=psb_err_invalid_cd_state_ call psb_errpush(info,name) goto 9999 endif if (present(data)) then data_ = data else data_ = psb_comm_halo_ end if call desc_a%get_list(data_,d_vidx,totxch,idxr,idxs,info) if (info /= psb_success_) then call psb_errpush(psb_err_internal_error_,name,a_err='psb_cd_get_list') goto 9999 end if call psi_swaptran(ictxt,icomm,flag,beta,y,d_vidx,totxch,idxs,idxr,work,info) if (info /= psb_success_) goto 9999 call psb_erractionrestore(err_act) return 9999 call psb_error_handler(ictxt,err_act) return end subroutine psi_sswaptran_multivect ! ! ! Subroutine: psi_stran_vidx_multivect ! Data exchange among processes. ! ! Takes care of Y an encapsulated multivector. Relies on the gather/scatter methods ! of multivectors. ! ! The real workhorse: the outer routine will only choose the index list ! this one takes the index list and does the actual exchange. ! ! ! subroutine psi_stran_vidx_multivect(iictxt,iicomm,flag,beta,y,idx,& & totxch,totsnd,totrcv,work,info) use psi_mod, psb_protect_name => psi_stran_vidx_multivect use psb_error_mod use psb_realloc_mod use psb_desc_mod use psb_penv_mod use psb_s_base_multivect_mod #ifdef MPI_MOD use mpi #endif implicit none #ifdef MPI_H include 'mpif.h' #endif integer(psb_ipk_), intent(in) :: iictxt,iicomm,flag integer(psb_ipk_), intent(out) :: info class(psb_s_base_multivect_type) :: y real(psb_spk_) :: beta real(psb_spk_), target :: work(:) class(psb_i_base_vect_type), intent(inout) :: idx integer(psb_ipk_), intent(in) :: totxch,totsnd, totrcv ! locals integer(psb_mpik_) :: ictxt, icomm, np, me,& & proc_to_comm, p2ptag, p2pstat(mpi_status_size), iret integer(psb_mpik_), allocatable :: prcid(:) integer(psb_ipk_) :: nesd, nerv,& & err_act, i, idx_pt, totsnd_, totrcv_,& & snd_pt, rcv_pt, pnti, n integer(psb_ipk_) :: ierr(5) logical :: swap_mpi, swap_sync, swap_send, swap_recv,& & albf,do_send,do_recv logical, parameter :: usersend=.false., debug=.false. character(len=20) :: name info=psb_success_ name='psi_swap_tran' call psb_erractionsave(err_act) ictxt = iictxt icomm = iicomm call psb_info(ictxt,me,np) if (np == -1) then info=psb_err_context_error_ call psb_errpush(info,name) goto 9999 endif n = y%get_ncols() swap_mpi = iand(flag,psb_swap_mpi_) /= 0 swap_sync = iand(flag,psb_swap_sync_) /= 0 swap_send = iand(flag,psb_swap_send_) /= 0 swap_recv = iand(flag,psb_swap_recv_) /= 0 do_send = swap_mpi .or. swap_sync .or. swap_send do_recv = swap_mpi .or. swap_sync .or. swap_recv totrcv_ = totrcv * n totsnd_ = totsnd * n call idx%sync() if (debug) write(*,*) me,'Internal buffer' if (do_send) then if (allocated(y%comid)) then if (any(y%comid /= mpi_request_null)) then ! ! Unfinished communication? Something is wrong.... ! info=psb_err_mpi_error_ ierr(1) = -2 call psb_errpush(info,name,i_err=ierr) goto 9999 end if end if if (debug) write(*,*) me,'do_send start' call y%new_buffer(ione*size(idx%v),info) call y%new_comid(totxch,info) y%comid = mpi_request_null call psb_realloc(totxch,prcid,info) ! First I post all the non blocking receives pnti = 1 snd_pt = totrcv_+1 rcv_pt = 1 p2ptag = psb_real_swap_tag do i=1, totxch proc_to_comm = idx%v(pnti+psb_proc_id_) nerv = idx%v(pnti+psb_n_elem_recv_) nesd = idx%v(pnti+nerv+psb_n_elem_send_) call psb_get_rank(prcid(i),ictxt,proc_to_comm) if ((nesd>0).and.(proc_to_comm /= me)) then if (debug) write(*,*) me,'Posting receive from',prcid(i),snd_pt call mpi_irecv(y%combuf(snd_pt),n*nesd,& & psb_mpi_r_spk_,prcid(i),& & p2ptag, icomm,y%comid(i,2),iret) end if rcv_pt = rcv_pt + n*nerv snd_pt = snd_pt + n*nesd pnti = pnti + nerv + nesd + 3 end do if (debug) write(*,*) me,' Gather ' ! ! Then gather for sending. ! pnti = 1 snd_pt = totrcv_+1 rcv_pt = 1 do i=1, totxch nerv = idx%v(pnti+psb_n_elem_recv_) nesd = idx%v(pnti+nerv+psb_n_elem_send_) idx_pt = 1+pnti+psb_n_elem_recv_ call y%gth(idx_pt,rcv_pt,nerv,idx) rcv_pt = rcv_pt + n*nerv snd_pt = snd_pt + n*nesd pnti = pnti + nerv + nesd + 3 end do ! ! Then wait for device ! call y%device_wait() if (debug) write(*,*) me,' isend' ! ! Then send ! pnti = 1 snd_pt = totrcv_+1 rcv_pt = 1 p2ptag = psb_real_swap_tag do i=1, totxch proc_to_comm = idx%v(pnti+psb_proc_id_) nerv = idx%v(pnti+psb_n_elem_recv_) nesd = idx%v(pnti+nerv+psb_n_elem_send_) idx_pt = 1+pnti+psb_n_elem_recv_ if ((nerv>0).and.(proc_to_comm /= me)) then call mpi_isend(y%combuf(rcv_pt),n*nerv,& & psb_mpi_r_spk_,prcid(i),& & p2ptag,icomm,y%comid(i,1),iret) end if if(iret /= mpi_success) then ierr(1) = iret info=psb_err_mpi_error_ call psb_errpush(info,name,i_err=ierr) goto 9999 end if rcv_pt = rcv_pt + n*nerv snd_pt = snd_pt + n*nesd pnti = pnti + nerv + nesd + 3 end do end if if (do_recv) then if (debug) write(*,*) me,' do_Recv' if (.not.allocated(y%comid)) then ! ! No matching send? Something is wrong.... ! info=psb_err_mpi_error_ ierr(1) = -2 call psb_errpush(info,name,i_err=ierr) goto 9999 end if call psb_realloc(totxch,prcid,info) if (debug) write(*,*) me,' wait' pnti = 1 snd_pt = totrcv_+1 rcv_pt = 1 p2ptag = psb_real_swap_tag do i=1, totxch proc_to_comm = idx%v(pnti+psb_proc_id_) nerv = idx%v(pnti+psb_n_elem_recv_) nesd = idx%v(pnti+nerv+psb_n_elem_send_) if (proc_to_comm /= me)then if (nerv>0) then call mpi_wait(y%comid(i,1),p2pstat,iret) if(iret /= mpi_success) then ierr(1) = iret info=psb_err_mpi_error_ call psb_errpush(info,name,i_err=ierr) goto 9999 end if end if if (nesd>0) then call mpi_wait(y%comid(i,2),p2pstat,iret) if(iret /= mpi_success) then ierr(1) = iret info=psb_err_mpi_error_ call psb_errpush(info,name,i_err=ierr) goto 9999 end if end if else if (proc_to_comm == me) then if (nesd /= nerv) then write(psb_err_unit,*) & & 'Fatal error in swapdata: mismatch on self send',& & nerv,nesd end if y%combuf(snd_pt:snd_pt+n*nesd-1) = y%combuf(rcv_pt:rcv_pt+n*nerv-1) end if rcv_pt = rcv_pt + n*nerv snd_pt = snd_pt + n*nesd pnti = pnti + nerv + nesd + 3 end do if (debug) write(*,*) me,' scatter' pnti = 1 snd_pt = totrcv_+1 rcv_pt = 1 do i=1, totxch proc_to_comm = idx%v(pnti+psb_proc_id_) nerv = idx%v(pnti+psb_n_elem_recv_) nesd = idx%v(pnti+nerv+psb_n_elem_send_) idx_pt = 1+pnti+nerv+psb_n_elem_send_ if (debug) write(0,*)me,' Received from: ',prcid(i),& & y%combuf(snd_pt:snd_pt+n*nesd-1) call y%sct(idx_pt,snd_pt,nesd,idx,beta) rcv_pt = rcv_pt + n*nerv snd_pt = snd_pt + n*nesd pnti = pnti + nerv + nesd + 3 end do ! ! Waited for com, cleanup comid ! y%comid = mpi_request_null ! ! Then wait for device ! if (debug) write(*,*) me,' wait' call y%device_wait() if (debug) write(*,*) me,' free buffer' call y%maybe_free_buffer(info) if (info == 0) call y%free_comid(info) if (info /= 0) then call psb_errpush(psb_err_alloc_dealloc_,name) goto 9999 end if if (debug) write(*,*) me,' done' end if call psb_erractionrestore(err_act) return 9999 call psb_error_handler(iictxt,err_act) return end subroutine psi_stran_vidx_multivect