Home
last modified time | relevance | path

Searched refs:nRemoteLeafRanks (Results 1 – 3 of 3) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu78 PetscInt i, nRemoteRootRanks, nRemoteLeafRanks; in PetscSFSetUp_Basic_NVSHMEM() local
89 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic_NVSHMEM()
91 bas->nRemoteLeafRanks = nRemoteLeafRanks; in PetscSFSetUp_Basic_NVSHMEM()
93 PetscCall(PetscMalloc2(nRemoteLeafRanks, &rootreqs, nRemoteRootRanks, &leafreqs)); in PetscSFSetUp_Basic_NVSHMEM()
97 stmp[2] = nRemoteLeafRanks; in PetscSFSetUp_Basic_NVSHMEM()
112 …for (i = 0; i < nRemoteLeafRanks; i++) PetscCallMPI(MPI_Send(&i, 1, MPIU_INT, bas->iranks[i + bas-… in PetscSFSetUp_Basic_NVSHMEM()
116 for (i = 0; i < nRemoteLeafRanks; i++) { in PetscSFSetUp_Basic_NVSHMEM()
133 PetscCall(PetscMalloc2(nRemoteLeafRanks, &bas->leafsigdisp, nRemoteLeafRanks, &bas->leafbufdisp)); in PetscSFSetUp_Basic_NVSHMEM()
134 …for (i = 0; i < nRemoteLeafRanks; i++) PetscCallMPI(MPIU_Irecv(&bas->leafsigdisp[i], 1, MPIU_INT, … in PetscSFSetUp_Basic_NVSHMEM()
136 PetscCallMPI(MPI_Waitall(nRemoteLeafRanks, rootreqs, MPI_STATUSES_IGNORE)); in PetscSFSetUp_Basic_NVSHMEM()
[all …]
/petsc/src/vec/is/sf/impls/basic/
H A Dsfbasic.h27 PetscMPIInt nRemoteLeafRanks; /* niranks - ndiranks */ member
H A Dsfbasic.c183 PetscMPIInt nRemoteRootRanks, nRemoteLeafRanks; in PetscSFSetUp_Basic() local
231 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic()
233 PetscCall(PetscMalloc2(nRemoteLeafRanks, &rootreqs, nRemoteRootRanks, &leafreqs)); in PetscSFSetUp_Basic()
246 PetscCallMPI(MPI_Waitall(nRemoteLeafRanks, rootreqs, MPI_STATUSES_IGNORE)); in PetscSFSetUp_Basic()
250 bas->nrootreqs = nRemoteLeafRanks; in PetscSFSetUp_Basic()