Lines Matching refs:locA
2406 Vec locA; in DMPlexComputeIntegral_Internal() local
2446 PetscCall(DMGetAuxiliaryVec(dm, NULL, 0, 0, &locA)); in DMPlexComputeIntegral_Internal()
2447 if (locA) { in DMPlexComputeIntegral_Internal()
2448 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeIntegral_Internal()
2511 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, c, NULL, &x)); in DMPlexComputeIntegral_Internal()
2513 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, c, NULL, &x)); in DMPlexComputeIntegral_Internal()
2719 Vec locA = NULL; in DMPlexComputeBdIntegral_Internal() local
2748 PetscCall(DMGetAuxiliaryVec(dm, NULL, 0, 0, &locA)); in DMPlexComputeBdIntegral_Internal()
2749 if (locA) { in DMPlexComputeBdIntegral_Internal()
2752 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeBdIntegral_Internal()
2772 PetscCall(PetscCalloc2(numFaces * totDim, &u, (locA ? (size_t)numFaces * totDimAux : 0), &a)); in DMPlexComputeBdIntegral_Internal()
2782 if (locA) { in DMPlexComputeBdIntegral_Internal()
2785 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdIntegral_Internal()
2787 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdIntegral_Internal()
3751 PetscErrorCode DMPlexGetCellFields(DM dm, IS cellIS, Vec locX, PeOp Vec locX_t, PeOp Vec locA, Pets… in DMPlexGetCellFields() argument
3764 if (locA) PetscValidHeaderSpecific(locA, VEC_CLASSID, 5); in DMPlexGetCellFields()
3773 if (locA) { in DMPlexGetCellFields()
3777 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexGetCellFields()
3788 if (locA) PetscCall(DMGetWorkArray(dm, numCells * totDimAux, MPIU_SCALAR, a)); in DMPlexGetCellFields()
3804 if (locA) { in DMPlexGetCellFields()
3807 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, subcell, NULL, &x)); in DMPlexGetCellFields()
3809 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, subcell, NULL, &x)); in DMPlexGetCellFields()
3813 if (locA) PetscCall(DMDestroy(&plexA)); in DMPlexGetCellFields()
3837 PetscErrorCode DMPlexRestoreCellFields(DM dm, IS cellIS, Vec locX, PeOp Vec locX_t, PeOp Vec locA, … in DMPlexRestoreCellFields() argument
3842 if (locA) PetscCall(DMRestoreWorkArray(dm, 0, MPIU_SCALAR, a)); in DMPlexRestoreCellFields()
3846 static PetscErrorCode DMPlexGetHybridCellFields(DM dm, IS cellIS, Vec locX, Vec locX_t, Vec locA, P… in DMPlexGetHybridCellFields() argument
3860 if (locA) PetscValidHeaderSpecific(locA, VEC_CLASSID, 5); in DMPlexGetHybridCellFields()
3871 if (locA) { in DMPlexGetHybridCellFields()
3875 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexGetHybridCellFields()
3887 if (locA) PetscCall(DMGetWorkArray(dm, numCells * totDimAux, MPIU_SCALAR, a)); in DMPlexGetHybridCellFields()
3958 if (locA) { in DMPlexGetHybridCellFields()
3963 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, subcell, &Nx, &x)); in DMPlexGetHybridCellFields()
3966 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, subcell, &Nx, &x)); in DMPlexGetHybridCellFields()
4400 Vec locA; in DMPlexComputeResidual_Patch_Internal() local
4419 PetscCall(DMGetAuxiliaryVec(dm, NULL, 0, 0, &locA)); in DMPlexComputeResidual_Patch_Internal()
4420 if (locA) { in DMPlexComputeResidual_Patch_Internal()
4421 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeResidual_Patch_Internal()
4486 PetscCall(DMPlexGetCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a)); in DMPlexComputeResidual_Patch_Internal()
4589 PetscCall(DMPlexRestoreCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a)); in DMPlexComputeResidual_Patch_Internal()
4926 Vec locA = NULL; in DMPlexComputeBdResidualSingleByKey() local
4935 PetscCall(DMGetAuxiliaryVec(dm, key.label, key.value, key.part, &locA)); in DMPlexComputeBdResidualSingleByKey()
4936 if (locA) { in DMPlexComputeBdResidualSingleByKey()
4939 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeBdResidualSingleByKey()
4966 …X_t ? (size_t)numFaces * totDim : 0), &u_t, numFaces * totDim, &elemVec, (locA ? (size_t)numFaces … in DMPlexComputeBdResidualSingleByKey()
4992 if (locA) { in DMPlexComputeBdResidualSingleByKey()
4996 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdResidualSingleByKey()
4998 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdResidualSingleByKey()
5190 Vec locA, cellGeometryFVM = NULL, faceGeometryFVM = NULL, locGrad = NULL; in DMPlexComputeResidualByKey() local
5215 PetscCall(DMGetAuxiliaryVec(dm, key.label, key.value, key.part, &locA)); in DMPlexComputeResidualByKey()
5216 if (locA) { in DMPlexComputeResidualByKey()
5218 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeResidualByKey()
5289 PetscCall(DMPlexGetCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a)); in DMPlexComputeResidualByKey()
5449 PetscCall(DMPlexRestoreCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a)); in DMPlexComputeResidualByKey()
5564 Vec locA[3] = {NULL, NULL, NULL}; in DMPlexComputeResidualHybridByKey() local
5605 PetscCall(DMGetAuxiliaryVec(dm, key[2].label, key[2].value, key[2].part, &locA[2])); in DMPlexComputeResidualHybridByKey()
5606 if (locA[2]) { in DMPlexComputeResidualHybridByKey()
5609 PetscCall(VecGetDM(locA[2], &dmAux[2])); in DMPlexComputeResidualHybridByKey()
5627 PetscCall(DMGetAuxiliaryVec(dm, key[c].label, key[c].value, key[c].part, &locA[c])); in DMPlexComputeResidualHybridByKey()
5628 …PetscCheck(locA[c], PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Must have auxiliary vector for (%p… in DMPlexComputeResidualHybridByKey()
5629 if (locA[c]) PetscCall(VecGetDM(locA[c], &dmAux[c])); in DMPlexComputeResidualHybridByKey()
5710 PetscCall(DMPlexGetHybridCellFields(dm, cellIS, locX, locX_t, locA[2], &u, &u_t, &a[2])); in DMPlexComputeResidualHybridByKey()
5711 PetscCall(DMPlexGetHybridFields(dm, dmAux, dsAux, cellIS, locA, PETSC_TRUE, a)); in DMPlexComputeResidualHybridByKey()
5835 PetscCall(DMPlexRestoreCellFields(dm, cellIS, locX, locX_t, locA[2], &u, &u_t, &a[2])); in DMPlexComputeResidualHybridByKey()
5836 PetscCall(DMPlexRestoreHybridFields(dm, dmAux, dsAux, cellIS, locA, PETSC_TRUE, a)); in DMPlexComputeResidualHybridByKey()
5914 Vec locA = NULL, tv; in DMPlexComputeBdJacobianSingleByLabel() local
5932 PetscCall(DMGetAuxiliaryVec(dm, label, values[0], 0, &locA)); in DMPlexComputeBdJacobianSingleByLabel()
5933 if (locA) { in DMPlexComputeBdJacobianSingleByLabel()
5936 PetscCall(VecGetDM(locA, &dmAux)); in DMPlexComputeBdJacobianSingleByLabel()
5969 …&elemMat, (hasPrec ? (size_t)numFaces * totDim * totDim : 0), &elemMatP, (locA ? (size_t)numFaces … in DMPlexComputeBdJacobianSingleByLabel()
5995 if (locA) { in DMPlexComputeBdJacobianSingleByLabel()
5998 PetscCall(DMPlexVecGetClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdJacobianSingleByLabel()
6000 PetscCall(DMPlexVecRestoreClosure(plexA, sectionAux, locA, subp, NULL, &x)); in DMPlexComputeBdJacobianSingleByLabel()
6642 Vec locA[3] = {NULL, NULL, NULL}; in DMPlexComputeJacobianHybridByKey() local
6688 PetscCall(DMGetAuxiliaryVec(dm, key[2].label, key[2].value, key[2].part, &locA[2])); in DMPlexComputeJacobianHybridByKey()
6689 if (locA[2]) { in DMPlexComputeJacobianHybridByKey()
6692 PetscCall(VecGetDM(locA[2], &dmAux[2])); in DMPlexComputeJacobianHybridByKey()
6712 PetscCall(DMGetAuxiliaryVec(dm, key[c].label, key[c].value, key[c].part, &locA[c])); in DMPlexComputeJacobianHybridByKey()
6713 if (locA[c]) PetscCall(VecGetDM(locA[c], &dmAux[c])); in DMPlexComputeJacobianHybridByKey()
6781 PetscCall(DMPlexGetHybridCellFields(dm, cellIS, locX, locX_t, locA[2], &u, &u_t, &a[2])); in DMPlexComputeJacobianHybridByKey()
6782 PetscCall(DMPlexGetHybridFields(dm, dmAux, dsAux, cellIS, locA, PETSC_TRUE, a)); in DMPlexComputeJacobianHybridByKey()
6947 PetscCall(DMPlexRestoreCellFields(dm, cellIS, locX, locX_t, locA[2], &u, &u_t, &a[2])); in DMPlexComputeJacobianHybridByKey()
6948 PetscCall(DMPlexRestoreHybridFields(dm, dmAux, dsAux, cellIS, locA, PETSC_TRUE, a)); in DMPlexComputeJacobianHybridByKey()