344   PetscInt nsrcranks, ndstranks, nLocallyAccessible = 0;  in PetscSFLinkGetDataBegin_NVSHMEM()  local
365     ndstranks = bas->nRemoteLeafRanks;  in PetscSFLinkGetDataBegin_NVSHMEM()
383     ndstranks = sf->nRemoteRootRanks;  in PetscSFLinkGetDataBegin_NVSHMEM()
395   if (ndstranks) {  in PetscSFLinkGetDataBegin_NVSHMEM()
396 …NvshmemSendSignals<<<(ndstranks + 255) / 256, 256, 0, link->remoteCommStream>>>(ndstranks, dstsig,…  in PetscSFLinkGetDataBegin_NVSHMEM()
488 __global__ static void WaitAndPutDataToRemotelyAccessible(PetscInt ndstranks, PetscMPIInt *dstranks…  in WaitAndPutDataToRemotelyAccessible()  argument
502 __global__ static void WaitSignalsFromLocallyAccessible(PetscInt ndstranks, PetscMPIInt *dstranks, …  in WaitSignalsFromLocallyAccessible()  argument
504   for (int i = 0; i < ndstranks; i++) {  in WaitSignalsFromLocallyAccessible()
518   PetscInt       ndstranks, nLocallyAccessible = 0;  in PetscSFLinkPutDataBegin_NVSHMEM()  local
529 …ndstranks = bas->nRemoteLeafRanks;                               /* number of (remote) leaf ranks …  in PetscSFLinkPutDataBegin_NVSHMEM()
542     ndstranks = sf->nRemoteRootRanks;  in PetscSFLinkPutDataBegin_NVSHMEM()
559   for (int i = 0; i < ndstranks; i++) {  in PetscSFLinkPutDataBegin_NVSHMEM()
564   if (nLocallyAccessible < ndstranks) {  in PetscSFLinkPutDataBegin_NVSHMEM()
565 …WaitAndPutDataToRemotelyAccessible<<<ndstranks, 1, 0, link->remoteCommStream>>>(ndstranks, dstrank…  in PetscSFLinkPutDataBegin_NVSHMEM()
571 …WaitSignalsFromLocallyAccessible<<<1, 1, 0, link->remoteCommStream>>>(ndstranks, dstranks_d, srcsi…  in PetscSFLinkPutDataBegin_NVSHMEM()
572     for (int i = 0; i < ndstranks; i++) {  in PetscSFLinkPutDataBegin_NVSHMEM()
587 __global__ static void PutDataEnd(PetscInt nsrcranks, PetscInt ndstranks, PetscMPIInt *dstranks, ui…  in PutDataEnd()  argument
596 …for (int i = 0; i < ndstranks; i++) nvshmemx_uint64_signal(dstsig + dstsigdisp[i], 1, dstranks[i])…  in PutDataEnd()
612   PetscInt       nsrcranks, ndstranks, *dstsigdisp;  in PetscSFLinkPutDataEnd_NVSHMEM()  local
618     ndstranks  = bas->nRemoteLeafRanks;  in PetscSFLinkPutDataEnd_NVSHMEM()
625     ndstranks  = sf->nRemoteRootRanks;  in PetscSFLinkPutDataEnd_NVSHMEM()
631   if (nsrcranks || ndstranks) {  in PetscSFLinkPutDataEnd_NVSHMEM()
632 …PutDataEnd<<<1, 1, 0, link->remoteCommStream>>>(nsrcranks, ndstranks, dstranks, dstsig, dstsigdisp…  in PetscSFLinkPutDataEnd_NVSHMEM()