Searched refs:nRemoteRootRanks (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 78 PetscInt i, nRemoteRootRanks, nRemoteLeafRanks; in PetscSFSetUp_Basic_NVSHMEM() local
88 nRemoteRootRanks = sf->nranks - sf->ndranks; in PetscSFSetUp_Basic_NVSHMEM()
90 sf->nRemoteRootRanks = nRemoteRootRanks; in PetscSFSetUp_Basic_NVSHMEM()
93 PetscCall(PetscMalloc2(nRemoteLeafRanks, &rootreqs, nRemoteRootRanks, &leafreqs)); in PetscSFSetUp_Basic_NVSHMEM()
95 stmp[0] = nRemoteRootRanks; in PetscSFSetUp_Basic_NVSHMEM()
110 PetscCall(PetscMalloc2(nRemoteRootRanks, &sf->rootsigdisp, nRemoteRootRanks, &sf->rootbufdisp)); in PetscSFSetUp_Basic_NVSHMEM()
111 …for (i = 0; i < nRemoteRootRanks; i++) PetscCallMPI(MPIU_Irecv(&sf->rootsigdisp[i], 1, MPIU_INT, s… in PetscSFSetUp_Basic_NVSHMEM()
113 PetscCallMPI(MPI_Waitall(nRemoteRootRanks, leafreqs, MPI_STATUSES_IGNORE)); in PetscSFSetUp_Basic_NVSHMEM()
115 …for (i = 0; i < nRemoteRootRanks; i++) PetscCallMPI(MPIU_Irecv(&sf->rootbufdisp[i], 1, MPIU_INT, s… in PetscSFSetUp_Basic_NVSHMEM()
120 PetscCallMPI(MPI_Waitall(nRemoteRootRanks, leafreqs, MPI_STATUSES_IGNORE)); in PetscSFSetUp_Basic_NVSHMEM()
[all …]
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.c | 183 PetscMPIInt nRemoteRootRanks, nRemoteLeafRanks; in PetscSFSetUp_Basic() local
202 nRemoteRootRanks = sf->nranks - sf->ndranks; in PetscSFSetUp_Basic()
203 …PetscCall(PetscCommBuildTwoSided(comm, 1, MPIU_INT, nRemoteRootRanks, PetscSafePointerPlusOffset(s… in PetscSFSetUp_Basic()
233 PetscCall(PetscMalloc2(nRemoteLeafRanks, &rootreqs, nRemoteRootRanks, &leafreqs)); in PetscSFSetUp_Basic()
247 PetscCallMPI(MPI_Waitall(nRemoteRootRanks, leafreqs, MPI_STATUSES_IGNORE)); in PetscSFSetUp_Basic()
249 sf->nleafreqs = nRemoteRootRanks; in PetscSFSetUp_Basic()
|
| /petsc/include/petsc/private/ |
| H A D | sfimpl.h | 120 PetscInt nRemoteRootRanks; /* nranks - ndranks */ member
|