Searched refs:dstsig (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 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
[all …]
|