Searched refs:rootbufdisp (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 65 PetscCall(PetscFree2(sf->rootsigdisp, sf->rootbufdisp)); in PetscSFReset_Basic_NVSHMEM()
110 PetscCall(PetscMalloc2(nRemoteRootRanks, &sf->rootsigdisp, nRemoteRootRanks, &sf->rootbufdisp)); in PetscSFSetUp_Basic_NVSHMEM()
115 …for (i = 0; i < nRemoteRootRanks; i++) PetscCallMPI(MPIU_Irecv(&sf->rootbufdisp[i], 1, MPIU_INT, s… in PetscSFSetUp_Basic_NVSHMEM()
127 …PetscCallCUDA(cudaMemcpyAsync(sf->rootbufdisp_d, sf->rootbufdisp, nRemoteRootRanks * sizeof(PetscI… in PetscSFSetUp_Basic_NVSHMEM()
360 …srcdisp_h = sf->rootbufdisp; /* for my i-th remote root rank, I will access its buf at offset roo… in PetscSFLinkGetDataBegin_NVSHMEM()
550 …dstdisp_h = sf->rootbufdisp; /* for my i-th remote root rank, I will access its root buf at offse… in PetscSFLinkPutDataBegin_NVSHMEM()
|
| /petsc/include/petsc/private/ |
| H A D | sfimpl.h | 125 …PetscInt *rootbufdisp; /* [nRemoteRootRanks]. For my i-th remote root rank, I will access its root… member
|