Lines Matching refs:ndiranks

89   nRemoteLeafRanks      = bas->niranks - bas->ndiranks;  in PetscSFSetUp_Basic_NVSHMEM()
112 …Ranks; i++) PetscCallMPI(MPI_Send(&i, 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm));   …  in PetscSFSetUp_Basic_NVSHMEM()
117     tmp = bas->ioffset[i + bas->ndiranks] - bas->ioffset[bas->ndiranks];  in PetscSFSetUp_Basic_NVSHMEM()
118 …PetscCallMPI(MPI_Send(&tmp, 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm)); /* Roots sen…  in PetscSFSetUp_Basic_NVSHMEM()
134 …lMPI(MPIU_Irecv(&bas->leafsigdisp[i], 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm, &roo…  in PetscSFSetUp_Basic_NVSHMEM()
138 …lMPI(MPIU_Irecv(&bas->leafbufdisp[i], 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm, &roo…  in PetscSFSetUp_Basic_NVSHMEM()
152 …PetscCallCUDA(cudaMemcpyAsync(bas->iranks_d, bas->iranks + bas->ndiranks, nRemoteLeafRanks * sizeo…  in PetscSFSetUp_Basic_NVSHMEM()
153 …PetscCallCUDA(cudaMemcpyAsync(bas->ioffset_d, bas->ioffset + bas->ndiranks, (nRemoteLeafRanks + 1)…  in PetscSFSetUp_Basic_NVSHMEM()
380     srcranks_h = bas->iranks + bas->ndiranks; /* my (remote) root ranks */  in PetscSFLinkGetDataBegin_NVSHMEM()
386 …dstdisp_h  = bas->ioffset + bas->ndiranks; /* offsets of the local root buf. Note dstdisp[0] is no…  in PetscSFLinkGetDataBegin_NVSHMEM()
533 …srcdisp_h = bas->ioffset + bas->ndiranks; /* offsets of rootbuf. srcdisp[0] is not necessarily zer…  in PetscSFLinkPutDataBegin_NVSHMEM()
539     dstranks_h = bas->iranks + bas->ndiranks; /* remote leaf ranks */  in PetscSFLinkPutDataBegin_NVSHMEM()