Lines Matching refs:bas
397 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
826 …if (PetscMemTypeDevice(link->rootmtype)) PetscCall(PetscLogGpuFlops(bas->rootbuflen[scope] * link-… in PetscSFLinkLogFlopsAfterUnpackRootData()
829 …PetscCall(PetscLogFlops(bas->rootbuflen[scope] * link->bs)); /* # of roots in buffer x # of scalar… in PetscSFLinkLogFlopsAfterUnpackRootData()
957 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkPackRootData() local
965 if (bas->rootbuflen[scope]) PetscCall(PetscSFLinkPackRootData_Private(sf, link, scope, rootdata)); in PetscSFLinkPackRootData()
987 …PetscSF_Basic *bas … in PetscSFLinkUnpackRootData_Private() local
994 PetscCall(PetscSFLinkGetUnpackAndOp(link, rootmtype, op, bas->rootdups[scope], &UnpackAndOp)); in PetscSFLinkUnpackRootData_Private()
1032 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkUnpackRootData() local
1036 …if (bas->rootbuflen[scope] && !link->rootdirect[scope]) PetscCall(PetscSFLinkUnpackRootData_Privat… in PetscSFLinkUnpackRootData()
1064 …PetscSF_Basic *bas … in PetscSFLinkFetchAndOpRemote() local
1071 if (bas->rootbuflen[PETSCSF_REMOTE]) { in PetscSFLinkFetchAndOpRemote()
1073 …PetscCall(PetscSFLinkGetFetchAndOp(link, rootmtype, op, bas->rootdups[PETSCSF_REMOTE], &FetchAndOp… in PetscSFLinkFetchAndOpRemote()
1086 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkScatterLocal() local
1119 …dstdups = (direction == PETSCSF_ROOT2LEAF) ? sf->leafdups[PETSCSF_LOCAL] : bas->rootdups[PETSCSF_… in PetscSFLinkScatterLocal()
1148 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFLinkFetchAndOpLocal() local
1154 if (!bas->rootbuflen[PETSCSF_LOCAL]) PetscFunctionReturn(PETSC_SUCCESS); in PetscSFLinkFetchAndOpLocal()
1161 …PetscCall(PetscSFLinkGetFetchAndOpLocal(link, rootmtype, op, bas->rootdups[PETSCSF_LOCAL], &FetchA… in PetscSFLinkFetchAndOpLocal()
1282 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFSetUpPackFields() local
1291 bas->rootstart[i] = 0; in PetscSFSetUpPackFields()
1292 bas->rootcontig[i] = PETSC_TRUE; in PetscSFSetUpPackFields()
1293 bas->rootdups[i] = PETSC_FALSE; in PetscSFSetUpPackFields()
1321 bas->rootbuflen[0] = bas->ioffset[bas->ndiranks]; in PetscSFSetUpPackFields()
1322 bas->rootbuflen[1] = bas->ioffset[bas->niranks] - bas->ioffset[bas->ndiranks]; in PetscSFSetUpPackFields()
1324 if (bas->rootbuflen[0]) bas->rootstart[0] = bas->irootloc[0]; in PetscSFSetUpPackFields()
1325 if (bas->rootbuflen[1]) bas->rootstart[1] = bas->irootloc[bas->ioffset[bas->ndiranks]]; in PetscSFSetUpPackFields()
1327 for (i = 0; i < bas->ioffset[bas->ndiranks]; i++) { in PetscSFSetUpPackFields()
1328 if (bas->irootloc[i] != bas->rootstart[0] + i) { in PetscSFSetUpPackFields()
1329 bas->rootcontig[0] = PETSC_FALSE; in PetscSFSetUpPackFields()
1333 for (i = bas->ioffset[bas->ndiranks], j = 0; i < bas->ioffset[bas->niranks]; i++, j++) { in PetscSFSetUpPackFields()
1334 if (bas->irootloc[i] != bas->rootstart[1] + j) { in PetscSFSetUpPackFields()
1335 bas->rootcontig[1] = PETSC_FALSE; in PetscSFSetUpPackFields()
1340 …if (!bas->rootcontig[0]) PetscCall(PetscSFCreatePackOpt(bas->ndiranks, bas->ioffset, bas->irootloc… in PetscSFSetUpPackFields()
1341 …if (!bas->rootcontig[1]) PetscCall(PetscSFCreatePackOpt(bas->niranks - bas->ndiranks, bas->ioffset… in PetscSFSetUpPackFields()
1348 …if (!bas->rootcontig[0] && !ismulti) PetscCall(PetscCheckDupsInt(bas->rootbuflen[0], bas->irootloc… in PetscSFSetUpPackFields()
1349 …if (!bas->rootcontig[1] && !ismulti) PetscCall(PetscCheckDupsInt(bas->rootbuflen[1], bas->irootloc… in PetscSFSetUpPackFields()
1356 PetscSF_Basic *bas = (PetscSF_Basic *)sf->data; in PetscSFResetPackFields() local
1362 PetscCall(PetscSFDestroyPackOpt(sf, PETSC_MEMTYPE_HOST, &bas->rootpackopt[i])); in PetscSFResetPackFields()
1365 PetscCall(PetscSFDestroyPackOpt(sf, PETSC_MEMTYPE_DEVICE, &bas->rootpackopt_d[i])); in PetscSFResetPackFields()