Lines Matching refs:dstsig
351 uint64_t *dstsig; in PetscSFLinkGetDataBegin_NVSHMEM() local
372 dstsig = link->leafRecvSig; in PetscSFLinkGetDataBegin_NVSHMEM()
390 dstsig = link->rootRecvSig; in PetscSFLinkGetDataBegin_NVSHMEM()
396 …<<<(ndstranks + 255) / 256, 256, 0, link->remoteCommStream>>>(ndstranks, dstsig, dstsigdisp_d, dst… in PetscSFLinkGetDataBegin_NVSHMEM()
402 …NvshmemWaitSignals<<<1, 1, 0, link->remoteCommStream>>>(nsrcranks, dstsig, 1, 0); /* wait the sign… in PetscSFLinkGetDataBegin_NVSHMEM()
587 …PetscInt nsrcranks, PetscInt ndstranks, PetscMPIInt *dstranks, uint64_t *dstsig, PetscInt *dstsigd… in PutDataEnd() argument
596 …for (int i = 0; i < ndstranks; i++) nvshmemx_uint64_signal(dstsig + dstsigdisp[i], 1, dstranks[i])… in PutDataEnd()
600 …nvshmem_uint64_wait_until_all(dstsig, nsrcranks, NULL /*no mask*/, NVSHMEM_CMP_EQ, 1); /* wait sig… in PutDataEnd()
601 for (int i = 0; i < nsrcranks; i++) dstsig[i] = 0; in PutDataEnd()
611 uint64_t *dstsig; in PetscSFLinkPutDataEnd_NVSHMEM() local
620 dstsig = link->leafRecvSig; /* I will set my leaf ranks's RecvSig */ in PetscSFLinkPutDataEnd_NVSHMEM()
627 dstsig = link->rootRecvSig; in PetscSFLinkPutDataEnd_NVSHMEM()
632 …PutDataEnd<<<1, 1, 0, link->remoteCommStream>>>(nsrcranks, ndstranks, dstranks, dstsig, dstsigdisp… in PetscSFLinkPutDataEnd_NVSHMEM()