| /petsc/src/ts/impls/explicit/ssp/ |
| H A D | ssp.c | 14 PetscInt nwork; member
24 if (ssp->nwork < n) { in TSSSPGetWorkVectors()
25 if (ssp->nwork > 0) PetscCall(VecDestroyVecs(ssp->nwork, &ssp->work)); in TSSSPGetWorkVectors()
27 ssp->nwork = n; in TSSSPGetWorkVectors()
219 if (ssp->work) PetscCall(VecDestroyVecs(ssp->nwork, &ssp->work)); in TSReset_SSP()
220 ssp->nwork = 0; in TSReset_SSP()
|
| /petsc/src/ksp/ksp/impls/gmres/pipefgmres/ |
| H A D | pipefgmres.c | 607 PetscInt nwork = pipefgmres->nwork_alloc; /* number of work vector chunks allocated */ in KSPPIPEFGMRESGetNewVectors() local
623 PetscCall(KSPCreateVecs(ksp, nalloc, &pipefgmres->user_work[nwork], 0, NULL)); in KSPPIPEFGMRESGetNewVectors()
624 …for (k = 0; k < nalloc; k++) pipefgmres->vecs[it + VEC_OFFSET + k] = pipefgmres->user_work[nwork][… in KSPPIPEFGMRESGetNewVectors()
626 pipefgmres->mwork_alloc[nwork] = nalloc; in KSPPIPEFGMRESGetNewVectors()
629 PetscCall(KSPCreateVecs(ksp, nalloc, &pipefgmres->prevecs_user_work[nwork], 0, NULL)); in KSPPIPEFGMRESGetNewVectors()
630 …for (k = 0; k < nalloc; k++) pipefgmres->prevecs[it + k] = pipefgmres->prevecs_user_work[nwork][k]; in KSPPIPEFGMRESGetNewVectors()
632 PetscCall(KSPCreateVecs(ksp, nalloc, &pipefgmres->zvecs_user_work[nwork], 0, NULL)); in KSPPIPEFGMRESGetNewVectors()
633 for (k = 0; k < nalloc; k++) pipefgmres->zvecs[it + k] = pipefgmres->zvecs_user_work[nwork][k]; in KSPPIPEFGMRESGetNewVectors()
|
| /petsc/src/ksp/ksp/impls/gmres/fgmres/ |
| H A D | fgmres.c | 372 PetscInt nwork = fgmres->nwork_alloc; /* number of work vector chunks allocated */ in KSPFGMRESGetNewVectors() local
388 PetscCall(KSPCreateVecs(ksp, nalloc, &fgmres->user_work[nwork], 0, NULL)); in KSPFGMRESGetNewVectors()
389 for (k = 0; k < nalloc; k++) fgmres->vecs[it + VEC_OFFSET + k] = fgmres->user_work[nwork][k]; in KSPFGMRESGetNewVectors()
391 fgmres->mwork_alloc[nwork] = nalloc; in KSPFGMRESGetNewVectors()
394 PetscCall(KSPCreateVecs(ksp, nalloc, &fgmres->prevecs_user_work[nwork], 0, NULL)); in KSPFGMRESGetNewVectors()
395 for (k = 0; k < nalloc; k++) fgmres->prevecs[it + k] = fgmres->prevecs_user_work[nwork][k]; in KSPFGMRESGetNewVectors()
|
| /petsc/include/petsc/private/ |
| H A D | pcbjkokkosimpl.h | 48 PetscInt nwork; member
|
| H A D | linesearchimpl.h | 44 PetscInt nwork; member
|
| H A D | snesimpl.h | 126 PetscInt nwork; member
|
| H A D | tsimpl.h | 324 PetscInt nwork; member
|
| H A D | kspimpl.h | 155 PetscInt nwork; member
|
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkos.kokkos.cxx | 539 const PetscInt nwork = jac->nwork, nBlk = jac->nBlocks; in PCApply_BJKOKKOS() local
612 if (nShareVec > nwork) nShareVec = nwork; in PCApply_BJKOKKOS()
613 else nGlobBVec = nwork - nShareVec; in PCApply_BJKOKKOS()
619 global_buff_words = jac->n * nwork; in PCApply_BJKOKKOS()
620 nGlobBVec = nwork; // not needed == fix in PCApply_BJKOKKOS()
887 jac->nwork = 7; in PCSetUp_BJKOKKOS()
892 jac->nwork = 10; in PCSetUp_BJKOKKOS()
898 jac->nwork = 0; in PCSetUp_BJKOKKOS()
1059 …abs tol = %e, div tol = %e, max it =%" PetscInt_FMT ", type = %s\n", jac->nwork, jac->ksp->rtol, j… in PCView_BJKOKKOS()
|
| /petsc/src/mat/impls/lrc/ |
| H A D | lrc.c | 41 PetscInt nwork; in MatMult_LRC_kernel() local
58 PetscCall(VecGetLocalSize(Na->work1, &nwork)); in MatMult_LRC_kernel()
59 …PetscCallMPI(MPIU_Allreduce(w1, w2, nwork, MPIU_SCALAR, MPIU_SUM, PetscObjectComm((PetscObject)N))… in MatMult_LRC_kernel()
|
| /petsc/src/ksp/ksp/impls/gmres/lgmres/ |
| H A D | lgmres.c | 510 PetscInt nwork = lgmres->nwork_alloc; /* number of work vector chunks allocated */ in KSPLGMRESGetNewVectors() local
525 PetscCall(KSPCreateVecs(ksp, nalloc, &lgmres->user_work[nwork], 0, NULL)); in KSPLGMRESGetNewVectors()
527 lgmres->mwork_alloc[nwork] = nalloc; in KSPLGMRESGetNewVectors()
529 for (k = 0; k < nalloc; k++) lgmres->vecs[it + VEC_OFFSET + k] = lgmres->user_work[nwork][k]; in KSPLGMRESGetNewVectors()
|
| /petsc/src/ksp/ksp/impls/bcgsl/ |
| H A D | bcgsl.c | 325 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPBCGSLSetXRes()
390 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPBCGSLSetPol()
432 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPBCGSLSetEll()
511 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPReset_BCGSL()
|
| /petsc/src/ksp/ksp/impls/gmres/ |
| H A D | gmres.c | 404 PetscInt nwork = gmres->nwork_alloc, k, nalloc; in KSPGMRESGetNewVectors() local
415 PetscCall(KSPCreateVecs(ksp, nalloc, &gmres->user_work[nwork], 0, NULL)); in KSPGMRESGetNewVectors()
417 gmres->mwork_alloc[nwork] = nalloc; in KSPGMRESGetNewVectors()
418 for (k = 0; k < nalloc; k++) gmres->vecs[it + VEC_OFFSET + k] = gmres->user_work[nwork][k]; in KSPGMRESGetNewVectors()
|
| /petsc/src/snes/linesearch/interface/ |
| H A D | linesearch.c | 266 PetscCall(VecDestroyVecs(linesearch->nwork, &linesearch->work)); in SNESLineSearchReset()
268 linesearch->nwork = 0; in SNESLineSearchReset()
1621 PetscErrorCode SNESLineSearchSetWorkVecs(SNESLineSearch linesearch, PetscInt nwork) in SNESLineSearchSetWorkVecs() argument
1625 PetscCall(VecDuplicateVecs(linesearch->vec_sol, nwork, &linesearch->work)); in SNESLineSearchSetWorkVecs()
|
| /petsc/src/snes/interface/ |
| H A D | snesut.c | 839 if (snes->work) PetscCall(VecDestroyVecs(snes->nwork, &snes->work)); in SNESSetWorkVecs()
840 snes->nwork = nw; in SNESSetWorkVecs()
844 PetscCall(VecDuplicateVecs(v, snes->nwork, &snes->work)); in SNESSetWorkVecs()
|
| H A D | snes.c | 1868 snes->nwork = 0; in SNESCreate()
3490 PetscCall(VecDestroyVecs(snes->nwork, &snes->work)); in SNESReset()
3495 snes->nwork = snes->nvwork = 0; in SNESReset()
|
| /petsc/src/ksp/ksp/impls/cg/ |
| H A D | cg.c | 83 PetscInt maxit = ksp->max_it, nwork = 3; in KSPSetUp_CG() local
87 if (cgP->singlereduction) nwork += 2; in KSPSetUp_CG()
88 PetscCall(KSPSetWorkVecs(ksp, nwork)); in KSPSetUp_CG()
|
| /petsc/src/snes/impls/gs/ |
| H A D | gssecant.c | 27 if (snes->nwork < 3) PetscCall(SNESSetWorkVecs(snes, 3)); in SNESComputeNGSDefaultSecant()
|
| /petsc/src/snes/impls/ms/ |
| H A D | ms.c | 367 PetscInt nwork = tab->nregisters + 1; // +1 because VecMAXPY() in SNESMSStep_3Sstar() in SNESSetUp_MS() local
371 PetscCall(SNESSetWorkVecs(snes, nwork)); in SNESSetUp_MS()
|
| /petsc/src/ksp/ksp/impls/gmres/dgmres/ |
| H A D | dgmres.c | 420 PetscInt nwork = dgmres->nwork_alloc, k, nalloc; in KSPDGMRESGetNewVectors() local
431 PetscCall(KSPCreateVecs(ksp, nalloc, &dgmres->user_work[nwork], 0, NULL)); in KSPDGMRESGetNewVectors()
433 dgmres->mwork_alloc[nwork] = nalloc; in KSPDGMRESGetNewVectors()
434 for (k = 0; k < nalloc; k++) dgmres->vecs[it + VEC_OFFSET + k] = dgmres->user_work[nwork][k]; in KSPDGMRESGetNewVectors()
|
| /petsc/src/ksp/ksp/interface/ |
| H A D | itcreate.c | 784 ksp->nwork = 0; in KSPCreate()
|
| H A D | iterativ.c | 1858 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPSetWorkVecs()
1859 ksp->nwork = nw; in KSPSetWorkVecs()
|
| /petsc/src/ksp/ksp/impls/fcg/pipefcg/ |
| H A D | pipefcg.c | 341 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); in KSPDestroy_PIPEFCG()
|
| /petsc/src/ksp/ksp/impls/gcr/pipegcr/ |
| H A D | pipegcr.c | 398 PetscCall(VecDestroyVecs(ksp->nwork, &ksp->work)); /* Destroy "standard" work vecs */ in KSPDestroy_PIPEGCR()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 1189 const auto nwork = nv * MDOT_WORKGROUP_NUM; in MDot_() local
1196 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), nwork, &d_results)); in MDot_()
1257 PetscCall(PetscLogGpuFlops(nwork)); in MDot_()
|