78   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()
138 …for (i = 0; i < nRemoteLeafRanks; i++) PetscCallMPI(MPIU_Irecv(&bas->leafbufdisp[i], 1, MPIU_INT, …  in PetscSFSetUp_Basic_NVSHMEM()
143   PetscCallMPI(MPI_Waitall(nRemoteLeafRanks, rootreqs, MPI_STATUSES_IGNORE));  in PetscSFSetUp_Basic_NVSHMEM()
145   PetscCallCUDA(cudaMalloc((void **)&bas->leafbufdisp_d, nRemoteLeafRanks * sizeof(PetscInt)));  in PetscSFSetUp_Basic_NVSHMEM()
146   PetscCallCUDA(cudaMalloc((void **)&bas->leafsigdisp_d, nRemoteLeafRanks * sizeof(PetscInt)));  in PetscSFSetUp_Basic_NVSHMEM()
147   PetscCallCUDA(cudaMalloc((void **)&bas->iranks_d, nRemoteLeafRanks * sizeof(PetscMPIInt)));  in PetscSFSetUp_Basic_NVSHMEM()
148   PetscCallCUDA(cudaMalloc((void **)&bas->ioffset_d, (nRemoteLeafRanks + 1) * sizeof(PetscInt)));  in PetscSFSetUp_Basic_NVSHMEM()
150 …PetscCallCUDA(cudaMemcpyAsync(bas->leafbufdisp_d, bas->leafbufdisp, nRemoteLeafRanks * sizeof(Pets…  in PetscSFSetUp_Basic_NVSHMEM()
151 …PetscCallCUDA(cudaMemcpyAsync(bas->leafsigdisp_d, bas->leafsigdisp, nRemoteLeafRanks * sizeof(Pets…  in PetscSFSetUp_Basic_NVSHMEM()
152 …PetscCallCUDA(cudaMemcpyAsync(bas->iranks_d, bas->iranks + bas->ndiranks, nRemoteLeafRanks * sizeo…  in PetscSFSetUp_Basic_NVSHMEM()
153 …PetscCallCUDA(cudaMemcpyAsync(bas->ioffset_d, bas->ioffset + bas->ndiranks, (nRemoteLeafRanks + 1)…  in PetscSFSetUp_Basic_NVSHMEM()
313     n   = bas->nRemoteLeafRanks;  in PetscSFLinkWaitSignalsOfCompletionOfGettingData_NVSHMEM()
365     ndstranks = bas->nRemoteLeafRanks;  in PetscSFLinkGetDataBegin_NVSHMEM()
375     nsrcranks = bas->nRemoteLeafRanks;  in PetscSFLinkGetDataBegin_NVSHMEM()
450     nsrcranks  = bas->nRemoteLeafRanks;  in PetscSFLinkGetDataEnd_NVSHMEM()
529 …ndstranks = bas->nRemoteLeafRanks;                               /* number of (remote) leaf ranks …  in PetscSFLinkPutDataBegin_NVSHMEM()
618     ndstranks  = bas->nRemoteLeafRanks;  in PetscSFLinkPutDataEnd_NVSHMEM()
623     nsrcranks = bas->nRemoteLeafRanks;  in PetscSFLinkPutDataEnd_NVSHMEM()
654     nsrcranks    = bas->nRemoteLeafRanks;  in PetscSFLinkSendSignalsToAllowPuttingData_NVSHMEM()