Searched refs:srcdisp_h (Results 1 – 1 of 1) sorted by relevance
347 PetscInt *srcdisp_h, *dstdisp_h; in PetscSFLinkGetDataBegin_NVSHMEM() local360 …srcdisp_h = sf->rootbufdisp; /* for my i-th remote root rank, I will access its buf at offset roo… in PetscSFLinkGetDataBegin_NVSHMEM()378 …srcdisp_h = bas->leafbufdisp; /* for my i-th remote root rank, I will access its buf at offset ro… in PetscSFLinkGetDataBegin_NVSHMEM()425 …_on_stream(dst + (dstdisp_h[i] - dstdisp_h[0]) * link->unitbytes, src + srcdisp_h[i] * link->unitb… in PetscSFLinkGetDataBegin_NVSHMEM()520 PetscInt *srcdisp_h, *dstdisp_h; in PetscSFLinkPutDataBegin_NVSHMEM() local533 …srcdisp_h = bas->ioffset + bas->ndiranks; /* offsets of rootbuf. srcdisp[0] is not necessarily zer… in PetscSFLinkPutDataBegin_NVSHMEM()546 srcdisp_h = sf->roffset + sf->ndranks; /* offsets of leafbuf */ in PetscSFLinkPutDataBegin_NVSHMEM()575 size_t nelems = (srcdisp_h[i + 1] - srcdisp_h[i]) * link->unitbytes; in PetscSFLinkPutDataBegin_NVSHMEM()577 …tmem_nbi_on_stream(dst + dstdisp_h[i] * link->unitbytes, src + (srcdisp_h[i] - srcdisp_h[0]) * lin… in PetscSFLinkPutDataBegin_NVSHMEM()