Home
last modified time | relevance | path

Searched refs:leafmtype_mpi (Results 1 – 6 of 6) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/
H A Dsfmpi.c9 const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; in PetscSFLinkFinishCommunication_Default() local
31 …PetscCallMPI(MPI_Wait(link->leafreqs[direction][leafmtype_mpi][leafdirect_mpi] + i, MPI_STATUS_IGN… in PetscSFLinkFinishCommunication_Default()
37 …qs) PetscCallMPI(MPI_Waitall(sf->nleafreqs, link->leafreqs[direction][leafmtype_mpi][leafdirect_mp… in PetscSFLinkFinishCommunication_Default()
74 PetscMemType rootmtype_mpi, leafmtype_mpi; /* mtypes seen by MPI */ in PetscSFLinkCreate_MPI() local
105 leafmtype_mpi = leafmtype; in PetscSFLinkCreate_MPI()
107 rootmtype_mpi = leafmtype_mpi = PETSC_MEMTYPE_HOST; in PetscSFLinkCreate_MPI()
111 leafdirect_mpi = leafdirect[PETSCSF_REMOTE] && (leafmtype_mpi == leafmtype) ? 1 : 0; in PetscSFLinkCreate_MPI()
213 if (PetscMemTypeDevice(leafmtype) && PetscMemTypeHost(leafmtype_mpi)) { in PetscSFLinkCreate_MPI()
236 link->leafmtype_mpi = leafmtype_mpi; in PetscSFLinkCreate_MPI()
H A Dsfbasic.c15 …const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; /* Us… in PetscSFLinkInitMPIRequests_Persistent_Basic() local
37 …if (sf->leafbuflen[PETSCSF_REMOTE] && !link->leafreqsinited[direction][leafmtype_mpi][leafdirect_m… in PetscSFLinkInitMPIRequests_Persistent_Basic()
43 …leafbuf[PETSCSF_REMOTE][leafmtype_mpi] + disp, cnt, unit, sf->ranks[i], link->tag, comm, link->lea… in PetscSFLinkInitMPIRequests_Persistent_Basic()
49 …leafbuf[PETSCSF_REMOTE][leafmtype_mpi] + disp, cnt, unit, sf->ranks[i], link->tag, comm, link->lea… in PetscSFLinkInitMPIRequests_Persistent_Basic()
52 link->leafreqsinited[direction][leafmtype_mpi][leafdirect_mpi] = PETSC_TRUE; in PetscSFLinkInitMPIRequests_Persistent_Basic()
106 …const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; /* Us… in PetscSFLinkStartCommunication_MPIX_Stream() local
135 …fbuf[PETSCSF_REMOTE][leafmtype_mpi] + disp, cnt, unit, sf->ranks[i], link->tag, stream_comm, link-… in PetscSFLinkStartCommunication_MPIX_Stream()
141 …fbuf[PETSCSF_REMOTE][leafmtype_mpi] + disp, cnt, unit, sf->ranks[i], link->tag, stream_comm, link-… in PetscSFLinkStartCommunication_MPIX_Stream()
151 const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; in PetscSFLinkFinishCommunication_MPIX_Stream() local
156 …PetscCallMPI(MPIX_Waitall_enqueue(sf->nleafreqs, link->leafreqs[direction][leafmtype_mpi][leafdire… in PetscSFLinkFinishCommunication_MPIX_Stream()
[all …]
H A Dsfpack.h181 …PetscMemType rootmtype_mpi, leafmtype_mpi; /* Mtypes of buffers passed to MPI. If use_gpu_aware_… member
268 …const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; /* me… in PetscSFLinkGetMPIBuffersAndRequests() local
275 if (leafbuf) *leafbuf = link->leafbuf[PETSCSF_REMOTE][leafmtype_mpi]; in PetscSFLinkGetMPIBuffersAndRequests()
277 if (leafreqs) *leafreqs = link->leafreqs[direction][leafmtype_mpi][leafdirect_mpi]; in PetscSFLinkGetMPIBuffersAndRequests()
324 …if (PetscMemTypeDevice(link->leafmtype) && PetscMemTypeHost(link->leafmtype_mpi) && sf->leafbuflen… in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
/petsc/src/vec/is/sf/impls/basic/allgather/
H A Dsfallgather.c75 …leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mpi, sf->leafbuflen[PETSCSF_REMOTE] * … in PetscSFReduceBegin_Allgather()
76 …if (rank == 0 && link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi] == leafbuf) leafbuf = MP… in PetscSFReduceBegin_Allgather()
79 …PetscCallMPI(MPI_Reduce(leafbuf, link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], count, … in PetscSFReduceBegin_Allgather()
80 …PetscCallMPI(MPIU_Iscatter(link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], recvcount, un… in PetscSFReduceBegin_Allgather()
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c120 …ern && rank == dat->bcast_root) PetscCall((*link->Memcpy)(link, link->leafmtype_mpi, leafbuf, link… in PetscSFBcastBegin_Allgatherv()
184 …leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mpi, sf->leafbuflen[PETSCSF_REMOTE] * … in PetscSFReduceBegin_Allgatherv()
186 …if (rank == 0 && link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi] == leafbuf) leafbuf = MP… in PetscSFReduceBegin_Allgatherv()
188 …PetscCallMPI(MPI_Reduce(leafbuf, link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], count, … in PetscSFReduceBegin_Allgatherv()
189 …PetscCallMPI(MPIU_Iscatterv(link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], dat->recvcou… in PetscSFReduceBegin_Allgatherv()
/petsc/src/vec/is/sf/impls/basic/neighbor/
H A Dsfneighbor.c122 …const PetscMemType rootmtype_mpi = link->rootmtype_mpi, leafmtype_mpi = link->leafmtype_mpi; /* Us… in PetscSFLinkInitMPIRequests_Persistent_Neighbor() local
125 …k->rootbuf[PETSCSF_REMOTE][rootmtype_mpi], *leafbuf = link->leafbuf[PETSCSF_REMOTE][leafmtype_mpi]; in PetscSFLinkInitMPIRequests_Persistent_Neighbor()