Home
last modified time | relevance | path

Searched refs:bas (Results 1 – 10 of 10) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu56 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFReset_Basic_NVSHMEM() local
59 PetscCall(PetscFree2(bas->leafsigdisp, bas->leafbufdisp)); in PetscSFReset_Basic_NVSHMEM()
60 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->leafbufdisp_d)); in PetscSFReset_Basic_NVSHMEM()
61 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->leafsigdisp_d)); in PetscSFReset_Basic_NVSHMEM()
62 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->iranks_d)); in PetscSFReset_Basic_NVSHMEM()
63 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->ioffset_d)); in PetscSFReset_Basic_NVSHMEM()
77 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFSetUp_Basic_NVSHMEM() local
89 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic_NVSHMEM()
91 bas->nRemoteLeafRanks = nRemoteLeafRanks; in PetscSFSetUp_Basic_NVSHMEM()
98 stmp[3] = bas->rootbuflen[PETSCSF_REMOTE]; in PetscSFSetUp_Basic_NVSHMEM()
[all …]
/petsc/src/vec/is/sf/impls/basic/
H A Dsfbasic.c8 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkInitMPIRequests_Persistent_Basic() local
19 …if (bas->rootbuflen[PETSCSF_REMOTE] && !link->rootreqsinited[direction][rootmtype_mpi][rootdirect_… in PetscSFLinkInitMPIRequests_Persistent_Basic()
25 …_Recv_init(link->rootbuf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[i], link->t… in PetscSFLinkInitMPIRequests_Persistent_Basic()
31 …_Send_init(link->rootbuf[PETSCSF_REMOTE][rootmtype_mpi] + disp, cnt, unit, bas->iranks[i], link->t… in PetscSFLinkInitMPIRequests_Persistent_Basic()
62 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkStartCommunication_Persistent_Basic() local
66 …rbuflen = (direction == PETSCSF_ROOT2LEAF) ? sf->leafbuflen[PETSCSF_REMOTE] : bas->rootbuflen[PETS… in PetscSFLinkStartCommunication_Persistent_Basic()
72 nrreqs = bas->nrootreqs; in PetscSFLinkStartCommunication_Persistent_Basic()
77 …sbuflen = (direction == PETSCSF_ROOT2LEAF) ? bas->rootbuflen[PETSCSF_REMOTE] : sf->leafbuflen[PETS… in PetscSFLinkStartCommunication_Persistent_Basic()
80 nsreqs = bas->nrootreqs; in PetscSFLinkStartCommunication_Persistent_Basic()
99 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkStartCommunication_MPIX_Stream() local
[all …]
H A Dsfmpi.c8 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkFinishCommunication_Default() local
22 for (PetscMPIInt i = 0; i < bas->nrootreqs; i++) { in PetscSFLinkFinishCommunication_Default()
23 …size_t size = (bas->ioffset[i + bas->ndiranks + 1] - bas->ioffset[i + bas->ndiranks]) * link->unit… in PetscSFLinkFinishCommunication_Default()
24 … Rank %6d (%16zu bytes) with MPI tag %10d ... ", rank, rootaction, bas->iranks[i + bas->ndiranks],… in PetscSFLinkFinishCommunication_Default()
36 …if (bas->nrootreqs) PetscCallMPI(MPI_Waitall(bas->nrootreqs, link->rootreqs[direction][rootmtype_m… in PetscSFLinkFinishCommunication_Default()
66 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkCreate_MPI() local
81 …rootdirect[i] = bas->rootcontig[i]; /* Pack roots… in PetscSFLinkCreate_MPI()
85 …rootdirect[i] = (bas->rootcontig[i] && op == MPI_REPLACE) ? PETSC_TRUE : PETSC_FALSE; /* Unpack ro… in PetscSFLinkCreate_MPI()
114 nrootreqs = bas->nrootreqs; in PetscSFLinkCreate_MPI()
118 for (p = &bas->avail; (link = *p); p = &link->next) { in PetscSFLinkCreate_MPI()
[all …]
H A Dsfpack.c397 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkDestroy() local
398 PetscInt i, nreqs = (bas->nrootreqs + sf->nleafreqs) * 8; in PetscSFLinkDestroy()
441 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkGetInUse() local
445 for (p = &bas->inuse; (link = *p); p = &link->next) { in PetscSFLinkGetInUse()
467 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkReclaim() local
473 link->next = bas->avail; in PetscSFLinkReclaim()
474 bas->avail = link; in PetscSFLinkReclaim()
483 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFSetErrorOnUnsupportedOverlap() local
492 for (p = &bas->inuse; (link = *p); p = &link->next) { in PetscSFSetErrorOnUnsupportedOverlap()
821 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkLogFlopsAfterUnpackRootData() local
[all …]
H A Dsfpack.h302 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()
[all …]
H A Dsfbasic.h42 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFGetRootInfo_Basic() local
45 if (nrootranks) *nrootranks = bas->niranks; in PetscSFGetRootInfo_Basic()
46 if (ndrootranks) *ndrootranks = bas->ndiranks; in PetscSFGetRootInfo_Basic()
47 if (rootranks) *rootranks = bas->iranks; in PetscSFGetRootInfo_Basic()
48 if (rootoffset) *rootoffset = bas->ioffset; in PetscSFGetRootInfo_Basic()
49 if (rootloc) *rootloc = bas->irootloc; in PetscSFGetRootInfo_Basic()
/petsc/src/vec/is/sf/impls/basic/alltoall/
H A Dsfalltoall.c95 PetscSF_Basic *bas; in PetscSFCreateEmbeddedRootSF_Alltoall() local
152 bas = (PetscSF_Basic *)esf->data; in PetscSFCreateEmbeddedRootSF_Alltoall()
153 PetscCall(PetscMalloc2(nroots, &bas->iranks, nroots + 1, &bas->ioffset)); in PetscSFCreateEmbeddedRootSF_Alltoall()
154 PetscCall(PetscMalloc1(nroots, &bas->irootloc)); in PetscSFCreateEmbeddedRootSF_Alltoall()
167 bas->niranks = nroots; in PetscSFCreateEmbeddedRootSF_Alltoall()
168 bas->ndiranks = ndiranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
169 bas->ioffset[0] = 0; in PetscSFCreateEmbeddedRootSF_Alltoall()
170 bas->itotal = nroots; in PetscSFCreateEmbeddedRootSF_Alltoall()
172 bas->iranks[i] = roots[i]; in PetscSFCreateEmbeddedRootSF_Alltoall()
173 bas->ioffset[i + 1] = i + 1; in PetscSFCreateEmbeddedRootSF_Alltoall()
[all …]
/petsc/src/ksp/ksp/tutorials/output/
H A Dex18_bas.out28 package used to perform factorization: bas
/petsc/src/vec/vec/utils/
H A Dvscat.c118 PetscSF_Basic *bas = NULL; in VecScatterRemap_Internal() local
167 bas = (PetscSF_Basic *)sf->data; in VecScatterRemap_Internal()
168 …for (i = 0; i < bas->ioffset[bas->niranks]; i++) bas->irootloc[i] = tomap[bas->irootloc[i] * bs] /… in VecScatterRemap_Internal()
171 for (i = 0; i < 2; i++) PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_DEVICE, bas->irootloc_d[i])); in VecScatterRemap_Internal()
/petsc/doc/manual/
H A Dksp.md2535 - ``bas``