Searched refs:srcsig (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 439 uint64_t *srcsig; in PetscSFLinkGetDataEnd_NVSHMEM() local
446 srcsig = link->rootSendSig; /* I want to set their root signal */ in PetscSFLinkGetDataEnd_NVSHMEM()
451 srcsig = link->leafSendSig; in PetscSFLinkGetDataEnd_NVSHMEM()
459 …<<<(nsrcranks + 511) / 512, 512, 0, link->remoteCommStream>>>(nsrcranks, srcsig, srcsigdisp, srcra… in PetscSFLinkGetDataEnd_NVSHMEM()
488 …r *dst, PetscInt *dstdisp, const char *src, PetscInt *srcdisp, uint64_t *srcsig, PetscInt unitbyte… in WaitAndPutDataToRemotelyAccessible() argument
495 nvshmem_uint64_wait_until(srcsig + bid, NVSHMEM_CMP_EQ, 0); /* Wait until the sig = 0 */ in WaitAndPutDataToRemotelyAccessible()
496 srcsig[bid] = 1; in WaitAndPutDataToRemotelyAccessible()
502 …FromLocallyAccessible(PetscInt ndstranks, PetscMPIInt *dstranks, uint64_t *srcsig, const char *dst) in WaitSignalsFromLocallyAccessible() argument
507 nvshmem_uint64_wait_until(srcsig + i, NVSHMEM_CMP_EQ, 0); /* Wait until the sig = 0 */ in WaitSignalsFromLocallyAccessible()
508 srcsig[i] = 1; in WaitSignalsFromLocallyAccessible()
[all …]
|