Searched refs:ndstranks (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 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()
[all …]
|