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 …]
|