Lines Matching refs:unitbytes
327 …nt *srcranks, const char *src, PetscInt *srcdisp, char *dst, PetscInt *dstdisp, PetscInt unitbytes) in GetDataFromRemotelyAccessible() argument
333 PetscInt nelems = (dstdisp[bid + 1] - dstdisp[bid]) * unitbytes; in GetDataFromRemotelyAccessible()
334 …nvshmem_getmem_nbi(dst + (dstdisp[bid] - dstdisp[0]) * unitbytes, src + srcdisp[bid] * unitbytes, … in GetDataFromRemotelyAccessible()
415 … link->remoteCommStream>>>(nsrcranks, srcranks_d, src, srcdisp_d, dst, dstdisp_d, link->unitbytes); in PetscSFLinkGetDataBegin_NVSHMEM()
424 size_t nelems = (dstdisp_h[i + 1] - dstdisp_h[i]) * link->unitbytes; in PetscSFLinkGetDataBegin_NVSHMEM()
425 …n_stream(dst + (dstdisp_h[i] - dstdisp_h[0]) * link->unitbytes, src + srcdisp_h[i] * link->unitbyt… in PetscSFLinkGetDataBegin_NVSHMEM()
488 … *dst, PetscInt *dstdisp, const char *src, PetscInt *srcdisp, uint64_t *srcsig, PetscInt unitbytes) in WaitAndPutDataToRemotelyAccessible() argument
494 PetscInt nelems = (srcdisp[bid + 1] - srcdisp[bid]) * unitbytes; in WaitAndPutDataToRemotelyAccessible()
497 …nvshmem_putmem_nbi(dst + dstdisp[bid] * unitbytes, src + (srcdisp[bid] - srcdisp[0]) * unitbytes, … in WaitAndPutDataToRemotelyAccessible()
565 …emoteCommStream>>>(ndstranks, dstranks_d, dst, dstdisp_d, src, srcdisp_d, srcsig, link->unitbytes); in PetscSFLinkPutDataBegin_NVSHMEM()
575 size_t nelems = (srcdisp_h[i + 1] - srcdisp_h[i]) * link->unitbytes; in PetscSFLinkPutDataBegin_NVSHMEM()
577 …em_nbi_on_stream(dst + dstdisp_h[i] * link->unitbytes, src + (srcdisp_h[i] - srcdisp_h[0]) * link-… in PetscSFLinkPutDataBegin_NVSHMEM()
769 …EMOTE][PETSC_MEMTYPE_DEVICE] = (char *)rootdata + bas->rootstart[PETSCSF_REMOTE] * link->unitbytes; in PetscSFLinkCreate_NVSHMEM()
771 …TYPE_DEVICE]) PetscCall(PetscNvshmemMalloc(bas->rootbuflen_rmax * link->unitbytes, (void **)&link-… in PetscSFLinkCreate_NVSHMEM()
776 …REMOTE][PETSC_MEMTYPE_DEVICE] = (char *)leafdata + sf->leafstart[PETSCSF_REMOTE] * link->unitbytes; in PetscSFLinkCreate_NVSHMEM()
778 …MTYPE_DEVICE]) PetscCall(PetscNvshmemMalloc(sf->leafbuflen_rmax * link->unitbytes, (void **)&link-… in PetscSFLinkCreate_NVSHMEM()