Lines Matching refs:pe
330 PetscMPIInt pe = srcranks[bid]; in GetDataFromRemotelyAccessible() local
332 if (!nvshmem_ptr(src, pe)) { in GetDataFromRemotelyAccessible()
334 …mem_nbi(dst + (dstdisp[bid] - dstdisp[0]) * unitbytes, src + srcdisp[bid] * unitbytes, nelems, pe); in GetDataFromRemotelyAccessible()
422 int pe = srcranks_h[i]; in PetscSFLinkGetDataBegin_NVSHMEM() local
423 if (nvshmem_ptr(src, pe)) { in PetscSFLinkGetDataBegin_NVSHMEM()
425 …isp_h[0]) * link->unitbytes, src + srcdisp_h[i] * link->unitbytes, nelems, pe, link->remoteCommStr… in PetscSFLinkGetDataBegin_NVSHMEM()
491 PetscMPIInt pe = dstranks[bid]; in WaitAndPutDataToRemotelyAccessible() local
493 if (!nvshmem_ptr(dst, pe)) { in WaitAndPutDataToRemotelyAccessible()
497 …mem_nbi(dst + dstdisp[bid] * unitbytes, src + (srcdisp[bid] - srcdisp[0]) * unitbytes, nelems, pe); in WaitAndPutDataToRemotelyAccessible()
505 int pe = dstranks[i]; in WaitSignalsFromLocallyAccessible() local
506 if (nvshmem_ptr(dst, pe)) { in WaitSignalsFromLocallyAccessible()
573 int pe = dstranks_h[i]; in PetscSFLinkPutDataBegin_NVSHMEM() local
574 … if (nvshmem_ptr(dst, pe)) { /* If return a non-null pointer, then <pe> is locally accessible */ in PetscSFLinkPutDataBegin_NVSHMEM()
577 …>unitbytes, src + (srcdisp_h[i] - srcdisp_h[0]) * link->unitbytes, nelems, pe, link->remoteCommStr… in PetscSFLinkPutDataBegin_NVSHMEM()