302   PetscSF_Basic *bas = (PetscSF_Basic *)sf->data;  in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()  local
306 …if (PetscMemTypeDevice(link->rootmtype) && PetscMemTypeHost(link->rootmtype_mpi) && bas->rootbufle…  in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
309     size_t count = bas->rootbuflen[PETSCSF_REMOTE] * link->unitbytes;  in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
342   PetscSF_Basic *bas = (PetscSF_Basic *)sf->data;  in PetscSFLinkSyncStreamBeforeCallMPI()  local
346 …if ((PetscMemTypeDevice(link->rootmtype) && bas->rootbuflen[PETSCSF_REMOTE]) || (PetscMemTypeDevic…  in PetscSFLinkSyncStreamBeforeCallMPI()
371   PetscSF_Basic *bas = (PetscSF_Basic *)sf->data;  in PetscSFLinkGetRootPackOptAndIndices()  local
375   *count   = bas->rootbuflen[scope];  in PetscSFLinkGetRootPackOptAndIndices()
376   *start   = bas->rootstart[scope];  in PetscSFLinkGetRootPackOptAndIndices()
385   if (!bas->rootcontig[scope]) {  in PetscSFLinkGetRootPackOptAndIndices()
386     offset = (scope == PETSCSF_LOCAL) ? 0 : bas->ioffset[bas->ndiranks];  in PetscSFLinkGetRootPackOptAndIndices()
388       *opt     = bas->rootpackopt[scope];  in PetscSFLinkGetRootPackOptAndIndices()
389       *indices = bas->irootloc + offset;  in PetscSFLinkGetRootPackOptAndIndices()
392       if (bas->rootpackopt[scope]) {  in PetscSFLinkGetRootPackOptAndIndices()
393         if (!bas->rootpackopt_d[scope]) {  in PetscSFLinkGetRootPackOptAndIndices()
394           PetscCall(PetscMalloc1(1, &bas->rootpackopt_d[scope]));  in PetscSFLinkGetRootPackOptAndIndices()
395 …PetscCall(PetscArraycpy(bas->rootpackopt_d[scope], bas->rootpackopt[scope], 1)); /* Make pointers …  in PetscSFLinkGetRootPackOptAndIndices()
396 …size = (bas->rootpackopt[scope]->n * 7 + 2) * sizeof(PetscInt);                  /* See comments a…  in PetscSFLinkGetRootPackOptAndIndices()
397 …PetscCall(PetscSFMalloc(sf, PETSC_MEMTYPE_DEVICE, size, (void **)&bas->rootpackopt_d[scope]->array…  in PetscSFLinkGetRootPackOptAndIndices()
398 …all((*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, bas->rootpackopt_d[scope]->array, PETSC_MEMTYPE_HO…  in PetscSFLinkGetRootPackOptAndIndices()
400         *opt = bas->rootpackopt_d[scope];  in PetscSFLinkGetRootPackOptAndIndices()
402         if (!bas->irootloc_d[scope]) {  in PetscSFLinkGetRootPackOptAndIndices()
403           size = bas->rootbuflen[scope] * sizeof(PetscInt);  in PetscSFLinkGetRootPackOptAndIndices()
404 …        PetscCall(PetscSFMalloc(sf, PETSC_MEMTYPE_DEVICE, size, (void **)&bas->irootloc_d[scope]));  in PetscSFLinkGetRootPackOptAndIndices()
405 …etscCall((*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, bas->irootloc_d[scope], PETSC_MEMTYPE_HOST, b…  in PetscSFLinkGetRootPackOptAndIndices()
407         *indices = bas->irootloc_d[scope];  in PetscSFLinkGetRootPackOptAndIndices()