Searched refs:roffset_d (Results 1 – 2 of 2) sorted by relevance
69 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->roffset_d)); in PetscSFReset_Basic_NVSHMEM()125 PetscCallCUDA(cudaMalloc((void **)&sf->roffset_d, (nRemoteRootRanks + 1) * sizeof(PetscInt))); in PetscSFSetUp_Basic_NVSHMEM()130 …PetscCallCUDA(cudaMemcpyAsync(sf->roffset_d, sf->roffset + sf->ndranks, (nRemoteRootRanks + 1) * s… in PetscSFSetUp_Basic_NVSHMEM()369 dstdisp_d = sf->roffset_d; in PetscSFLinkGetDataBegin_NVSHMEM()547 srcdisp_d = sf->roffset_d; in PetscSFLinkPutDataBegin_NVSHMEM()
130 PetscInt *roffset_d; /* Copy of the remote part of roffset[] on device */ member