Lines Matching refs:ndranks
88 nRemoteRootRanks = sf->nranks - sf->ndranks; in PetscSFSetUp_Basic_NVSHMEM()
111 …scCallMPI(MPIU_Irecv(&sf->rootsigdisp[i], 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm, &lea… in PetscSFSetUp_Basic_NVSHMEM()
115 …scCallMPI(MPIU_Irecv(&sf->rootbufdisp[i], 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm, &lea… in PetscSFSetUp_Basic_NVSHMEM()
129 …PetscCallCUDA(cudaMemcpyAsync(sf->ranks_d, sf->ranks + sf->ndranks, nRemoteRootRanks * sizeof(Pets… in PetscSFSetUp_Basic_NVSHMEM()
130 …PetscCallCUDA(cudaMemcpyAsync(sf->roffset_d, sf->roffset + sf->ndranks, (nRemoteRootRanks + 1) * s… in PetscSFSetUp_Basic_NVSHMEM()
135 …moteRootRanks; i++) PetscCallMPI(MPI_Send(&i, 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm)); in PetscSFSetUp_Basic_NVSHMEM()
140 tmp = sf->roffset[i + sf->ndranks] - sf->roffset[sf->ndranks]; in PetscSFSetUp_Basic_NVSHMEM()
141 PetscCallMPI(MPI_Send(&tmp, 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm)); in PetscSFSetUp_Basic_NVSHMEM()
362 srcranks_h = sf->ranks + sf->ndranks; /* my (remote) root ranks */ in PetscSFLinkGetDataBegin_NVSHMEM()
368 …dstdisp_h = sf->roffset + sf->ndranks; /* offsets of the local leaf buf. Note dstdisp[0] is not n… in PetscSFLinkGetDataBegin_NVSHMEM()
546 srcdisp_h = sf->roffset + sf->ndranks; /* offsets of leafbuf */ in PetscSFLinkPutDataBegin_NVSHMEM()
552 dstranks_h = sf->ranks + sf->ndranks; /* remote root ranks */ in PetscSFLinkPutDataBegin_NVSHMEM()