Searched refs:dstranks (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 488 __global__ static void WaitAndPutDataToRemotelyAccessible(PetscInt ndstranks, PetscMPIInt *dstranks… in WaitAndPutDataToRemotelyAccessible() argument
491 PetscMPIInt pe = dstranks[bid]; in WaitAndPutDataToRemotelyAccessible()
502 __global__ static void WaitSignalsFromLocallyAccessible(PetscInt ndstranks, PetscMPIInt *dstranks, … in WaitSignalsFromLocallyAccessible() argument
505 int pe = dstranks[i]; in WaitSignalsFromLocallyAccessible()
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()
610 PetscMPIInt *dstranks; in PetscSFLinkPutDataEnd_NVSHMEM() local
619 dstranks = bas->iranks_d; /* leaf ranks */ in PetscSFLinkPutDataEnd_NVSHMEM()
626 dstranks = sf->ranks_d; in PetscSFLinkPutDataEnd_NVSHMEM()
632 …PutDataEnd<<<1, 1, 0, link->remoteCommStream>>>(nsrcranks, ndstranks, dstranks, dstsig, dstsigdisp… in PetscSFLinkPutDataEnd_NVSHMEM()
|