Searched refs:nsrcranks (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 327 __global__ static void GetDataFromRemotelyAccessible(PetscInt nsrcranks, PetscMPIInt *srcranks, con… in GetDataFromRemotelyAccessible() argument
344 PetscInt nsrcranks, ndstranks, nLocallyAccessible = 0; in PetscSFLinkGetDataBegin_NVSHMEM() local
357 nsrcranks = sf->nRemoteRootRanks; in PetscSFLinkGetDataBegin_NVSHMEM()
375 nsrcranks = bas->nRemoteLeafRanks; in PetscSFLinkGetDataBegin_NVSHMEM()
401 if (nsrcranks) { in PetscSFLinkGetDataBegin_NVSHMEM()
402 …NvshmemWaitSignals<<<1, 1, 0, link->remoteCommStream>>>(nsrcranks, dstsig, 1, 0); /* wait the sign… in PetscSFLinkGetDataBegin_NVSHMEM()
409 for (int i = 0; i < nsrcranks; i++) { in PetscSFLinkGetDataBegin_NVSHMEM()
414 if (nLocallyAccessible < nsrcranks) { in PetscSFLinkGetDataBegin_NVSHMEM()
415 …GetDataFromRemotelyAccessible<<<nsrcranks, 1, 0, link->remoteCommStream>>>(nsrcranks, srcranks_d, … in PetscSFLinkGetDataBegin_NVSHMEM()
421 for (int i = 0; i < nsrcranks; i++) { in PetscSFLinkGetDataBegin_NVSHMEM()
[all …]
|