Searched refs:rootbufdisp_d (Results 1 – 2 of 2) sorted by relevance
66 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->rootbufdisp_d)); in PetscSFReset_Basic_NVSHMEM()122 PetscCallCUDA(cudaMalloc((void **)&sf->rootbufdisp_d, nRemoteRootRanks * sizeof(PetscInt))); in PetscSFSetUp_Basic_NVSHMEM()127 …PetscCallCUDA(cudaMemcpyAsync(sf->rootbufdisp_d, sf->rootbufdisp, nRemoteRootRanks * sizeof(PetscI… in PetscSFSetUp_Basic_NVSHMEM()361 srcdisp_d = sf->rootbufdisp_d; in PetscSFLinkGetDataBegin_NVSHMEM()551 dstdisp_d = sf->rootbufdisp_d; in PetscSFLinkPutDataBegin_NVSHMEM()
127 PetscInt *rootbufdisp_d; member