Home
last modified time | relevance | path

Searched refs:rootmtype_mpi (Results 1 – 5 of 5) 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
25 …PetscCallMPI(MPI_Wait(link->rootreqs[direction][rootmtype_mpi][rootdirect_mpi] + i, MPI_STATUS_IGN… in PetscSFLinkFinishCommunication_Default()
36 …s) PetscCallMPI(MPI_Waitall(bas->nrootreqs, link->rootreqs[direction][rootmtype_mpi][rootdirect_mp… in PetscSFLinkFinishCommunication_Default()
74 PetscMemType rootmtype_mpi, leafmtype_mpi; /* mtypes seen by MPI */ in PetscSFLinkCreate_MPI() local
104 rootmtype_mpi = rootmtype; in PetscSFLinkCreate_MPI()
107 rootmtype_mpi = leafmtype_mpi = PETSC_MEMTYPE_HOST; in PetscSFLinkCreate_MPI()
110 rootdirect_mpi = rootdirect[PETSCSF_REMOTE] && (rootmtype_mpi == rootmtype) ? 1 : 0; in PetscSFLinkCreate_MPI()
209 if (PetscMemTypeDevice(rootmtype) && PetscMemTypeHost(rootmtype_mpi)) { in PetscSFLinkCreate_MPI()
235 link->rootmtype_mpi = rootmtype_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
19 …if (bas->rootbuflen[PETSCSF_REMOTE] && !link->rootreqsinited[direction][rootmtype_mpi][rootdirect_… in PetscSFLinkInitMPIRequests_Persistent_Basic()
25 …ootbuf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[i], link->tag, comm, link->ro… in PetscSFLinkInitMPIRequests_Persistent_Basic()
31 …ootbuf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[i], link->tag, comm, link->ro… in PetscSFLinkInitMPIRequests_Persistent_Basic()
34 link->rootreqsinited[direction][rootmtype_mpi][rootdirect_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
116 …buf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[i], link->tag, stream_comm, link… in PetscSFLinkStartCommunication_MPIX_Stream()
123 …buf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[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
155 …PetscCallMPI(MPIX_Waitall_enqueue(bas->nrootreqs, link->rootreqs[direction][rootmtype_mpi][rootdir… 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
274 if (rootbuf) *rootbuf = link->rootbuf[PETSCSF_REMOTE][rootmtype_mpi]; in PetscSFLinkGetMPIBuffersAndRequests()
276 if (rootreqs) *rootreqs = link->rootreqs[direction][rootmtype_mpi][rootdirect_mpi]; in PetscSFLinkGetMPIBuffersAndRequests()
306 …if (PetscMemTypeDevice(link->rootmtype) && PetscMemTypeHost(link->rootmtype_mpi) && bas->rootbufle… in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
/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
124 MPI_Request *req = link->rootreqs[direction][rootmtype_mpi][rootdirect_mpi]; in PetscSFLinkInitMPIRequests_Persistent_Neighbor()
125 …void *rootbuf = link->rootbuf[PETSCSF_REMOTE][rootmtype_mpi], *leafbuf = link->leafbu… in PetscSFLinkInitMPIRequests_Persistent_Neighbor()
131 if (!link->rootreqsinited[direction][rootmtype_mpi][rootdirect_mpi]) { in PetscSFLinkInitMPIRequests_Persistent_Neighbor()
138 link->rootreqsinited[direction][rootmtype_mpi][rootdirect_mpi] = PETSC_TRUE; in PetscSFLinkInitMPIRequests_Persistent_Neighbor()
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c120 …) PetscCall((*link->Memcpy)(link, link->leafmtype_mpi, leafbuf, link->rootmtype_mpi, rootbuf, (siz… in PetscSFBcastBegin_Allgatherv()