Searched refs:ranks_d (Results 1 – 2 of 2) sorted by relevance
68 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->ranks_d)); in PetscSFReset_Basic_NVSHMEM()124 PetscCallCUDA(cudaMalloc((void **)&sf->ranks_d, nRemoteRootRanks * sizeof(PetscMPIInt))); in PetscSFSetUp_Basic_NVSHMEM()129 …PetscCallCUDA(cudaMemcpyAsync(sf->ranks_d, sf->ranks + sf->ndranks, nRemoteRootRanks * sizeof(Pets… in PetscSFSetUp_Basic_NVSHMEM()363 srcranks_d = sf->ranks_d; in PetscSFLinkGetDataBegin_NVSHMEM()388 dstranks_d = sf->ranks_d; /* my (remote) root ranks */ in PetscSFLinkGetDataBegin_NVSHMEM()448 srcranks = sf->ranks_d; /* ranks of the n root ranks */ in PetscSFLinkGetDataEnd_NVSHMEM()553 dstranks_d = sf->ranks_d; in PetscSFLinkPutDataBegin_NVSHMEM()626 dstranks = sf->ranks_d; in PetscSFLinkPutDataEnd_NVSHMEM()652 srcranks_d = sf->ranks_d; /* ranks of the n root ranks */ in PetscSFLinkSendSignalsToAllowPuttingData_NVSHMEM()
129 PetscMPIInt *ranks_d; /* Copy of the remote part of (root) ranks[] on device */ member