Lines Matching refs:totDim
2410 PetscInt Nf, totDim, *uOff, *uOff_x, numConstants; in DMPlexComputeIntegral_Internal() local
2440 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeIntegral_Internal()
2457 PetscCall(PetscCalloc1(numCells * totDim, &u)); in DMPlexComputeIntegral_Internal()
2508 for (i = 0; i < totDim; ++i) u[c * totDim + i] = x[i]; in DMPlexComputeIntegral_Internal()
2545 …PetscCall(PetscFEIntegrate(prob, f, Nr, chunkGeom, &u[offset * totDim], probAux, PetscSafePointerP… in DMPlexComputeIntegral_Internal()
2560 …obj_func(dim, Nf, NfAux, uOff, uOff_x, &u[totDim * c + foff], NULL, u_x, aOff, NULL, PetscSafePoin… in DMPlexComputeIntegral_Internal()
2721 PetscInt Nf, totDim, *uOff, *uOff_x; in DMPlexComputeBdIntegral_Internal() local
2743 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeBdIntegral_Internal()
2772 PetscCall(PetscCalloc2(numFaces * totDim, &u, (locA ? (size_t)numFaces * totDimAux : 0), &a)); in DMPlexComputeBdIntegral_Internal()
2780 for (PetscInt i = 0; i < totDim; ++i) u[face * totDim + i] = x[i]; in DMPlexComputeBdIntegral_Internal()
2822 …(prob, field, funcs[field], chunkSize, chunkGeom, &u[chunk * chunkSize * totDim], probAux, PetscSa… in DMPlexComputeBdIntegral_Internal()
2826 …PetscCall(PetscFEIntegrateBd(prob, field, funcs[field], Nr, chunkGeom, &u[offset * totDim], probAu… in DMPlexComputeBdIntegral_Internal()
3183 PetscInt dim, Nf, field, totDim, cStart, cEnd, cell, ccell, s; in DMPlexComputeInterpolatorGeneral() local
3198 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeInterpolatorGeneral()
3199 PetscCall(PetscMalloc1(totDim, &elemMat)); in DMPlexComputeInterpolatorGeneral()
3239 …== totDim, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of fine indices %" PetscInt_FMT " != %" P… in DMPlexComputeInterpolatorGeneral()
3299 …ndices == totDim, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of element matrix columns %" PetscInt_F… in DMPlexComputeInterpolatorGeneral()
3356 PetscInt dim, Nf, field, totDim, cStart, cEnd, cell, ccell; in DMPlexComputeMassMatrixGeneral() local
3370 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeMassMatrixGeneral()
3371 PetscCall(PetscMalloc1(totDim, &elemMat)); in DMPlexComputeMassMatrixGeneral()
3758 PetscInt cStart, cEnd, numCells, totDim, totDimAux, c; in DMPlexGetCellFields() local
3772 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexGetCellFields()
3785 PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, u)); in DMPlexGetCellFields()
3786 if (locX_t) PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, u_t)); in DMPlexGetCellFields()
3797 for (i = 0; i < totDim; ++i) ul[cind * totDim + i] = x[i]; in DMPlexGetCellFields()
3801 for (i = 0; i < totDim; ++i) ul_t[cind * totDim + i] = x_t[i]; in DMPlexGetCellFields()
3853 PetscInt cStart, cEnd, numCells, c, totDim, totDimAux, Nf, f; in DMPlexGetHybridCellFields() local
3870 PetscCall(PetscDSGetTotalDimension(dsIn, &totDim)); in DMPlexGetHybridCellFields()
3882 PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, u)); in DMPlexGetHybridCellFields()
3883 if (locX_t) PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, u_t)); in DMPlexGetHybridCellFields()
3896 …PetscScalar *ul = &(*u)[cind * totDim], *ul_t = PetscSafePointerPlusOffset(*u_t, cind * totDim); in DMPlexGetHybridCellFields()
3956 …totDim, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Closure size %" PetscInt_FMT " for cell %" PetscIn… in DMPlexGetHybridCellFields()
4405 PetscInt Nf, f, totDim, totDimAux, numChunks, cellChunkSize, chunk, fStart, fEnd; in DMPlexComputeResidual_Patch_Internal() local
4418 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeResidual_Patch_Internal()
4487 PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVec)); in DMPlexComputeResidual_Patch_Internal()
4488 PetscCall(PetscArrayzero(elemVec, numCells * totDim)); in DMPlexComputeResidual_Patch_Internal()
4525 …eom, &u[offset * totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, &a[offset * t… in DMPlexComputeResidual_Patch_Internal()
4543 … if (mesh->printFEM > 1) PetscCall(DMPrintCellVector(cell, name, totDim, &elemVec[cind * totDim])); in DMPlexComputeResidual_Patch_Internal()
4550 … PetscCall(DMPlexVecSetClosure(dm, section, locF, cell, &elemVec[cind * totDim], ADD_ALL_VALUES)); in DMPlexComputeResidual_Patch_Internal()
4590 PetscCall(DMRestoreWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVec)); in DMPlexComputeResidual_Patch_Internal()
4634 … fieldJ, maxDegree, numCells, cStart, cEnd, numChunks, chunkSize, chunk, totDim, totDimAux = 0, sz… in DMPlexComputeJacobian_Patch_Internal() local
4668 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeJacobian_Patch_Internal()
4691 …+ (X_t ? 1 : 0)) * totDim + (dmAux ? 1 : 0) * totDimAux + ((hasJac ? 1 : 0) + (hasPrec ? 1 : 0) + … in DMPlexComputeJacobian_Patch_Internal()
4695 u = X ? (sz = chunkSize * totDim, off += sz, work + off - sz) : NULL; in DMPlexComputeJacobian_Patch_Internal()
4696 u_t = X_t ? (sz = chunkSize * totDim, off += sz, work + off - sz) : NULL; in DMPlexComputeJacobian_Patch_Internal()
4698 elemMat = hasJac ? (sz = chunkSize * totDim * totDim, off += sz, work + off - sz) : NULL; in DMPlexComputeJacobian_Patch_Internal()
4699 elemMatP = hasPrec ? (sz = chunkSize * totDim * totDim, off += sz, work + off - sz) : NULL; in DMPlexComputeJacobian_Patch_Internal()
4700 elemMatD = hasDyn ? (sz = chunkSize * totDim * totDim, off += sz, work + off - sz) : NULL; in DMPlexComputeJacobian_Patch_Internal()
4732 for (i = 0; i < totDim; ++i) u[c * totDim + i] = x[i]; in DMPlexComputeJacobian_Patch_Internal()
4737 for (i = 0; i < totDim; ++i) u_t[c * totDim + i] = x_t[i]; in DMPlexComputeJacobian_Patch_Internal()
4763 for (c = 0; c < chunkSize; ++c, eOffset += totDim * totDim) { in DMPlexComputeJacobian_Patch_Internal()
4766 if (hasJac) elemMat[eOffset + i * totDim + i] = 1.0; in DMPlexComputeJacobian_Patch_Internal()
4767 if (hasPrec) elemMatP[eOffset + i * totDim + i] = 1.0; in DMPlexComputeJacobian_Patch_Internal()
4774 for (c = 0; c < chunkSize * totDim * totDim; ++c) elemMat[c] += X_tShift * elemMatD[c]; in DMPlexComputeJacobian_Patch_Internal()
4780 …(hasJac) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMat[(c - cStart) * totDim * … in DMPlexComputeJacobian_Patch_Internal()
4781 …sPrec) PetscCall(DMPrintCellMatrix(cell, nameP, totDim, totDim, &elemMatP[(c - cStart) * totDim * … in DMPlexComputeJacobian_Patch_Internal()
4783 …lobalSection, mesh->useMatClPerm, Jac, cell, &elemMat[(c - cStart) * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobian_Patch_Internal()
4784 …globalSection, mesh->useMatClPerm, JP, cell, &elemMat[(c - cStart) * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobian_Patch_Internal()
4792 …X_t ? 1 : 0) + (dmAux ? 1 : 0)) * totDim + ((hasJac ? 1 : 0) + (hasPrec ? 1 : 0) + (hasDyn ? 1 : 0… in DMPlexComputeJacobian_Patch_Internal()
4928 PetscInt totDim, totDimAux = 0; in DMPlexComputeBdResidualSingleByKey() local
4934 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeBdResidualSingleByKey()
4966 …PetscCall(PetscMalloc4(numFaces * totDim, &u, (locX_t ? (size_t)numFaces * totDim : 0), &u_t, numF… in DMPlexComputeBdResidualSingleByKey()
4985 for (i = 0; i < totDim; ++i) u[face * totDim + i] = x[i]; in DMPlexComputeBdResidualSingleByKey()
4989 for (i = 0; i < totDim; ++i) u_t[face * totDim + i] = x[i]; in DMPlexComputeBdResidualSingleByKey()
5001 PetscCall(PetscArrayzero(elemVec, numFaces * totDim)); in DMPlexComputeBdResidualSingleByKey()
5026 … * totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, PetscSafePointerPlusOffset(… in DMPlexComputeBdResidualSingleByKey()
5032 …if (mesh->printFEM > 1) PetscCall(DMPrintCellVector(point, name, totDim, &elemVec[face * totDim])); in DMPlexComputeBdResidualSingleByKey()
5034 …PetscCall(DMPlexVecSetClosure(plex, NULL, locF, support[0], &elemVec[face * totDim], ADD_ALL_VALUE… in DMPlexComputeBdResidualSingleByKey()
5195 …PetscInt Nf, f, totDim, totDimAux, numChunks, cellChunkSize, faceChunkSize, chunk, fStart, … in DMPlexComputeResidualByKey() local
5214 PetscCall(PetscDSGetTotalDimension(ds, &totDim)); in DMPlexComputeResidualByKey()
5290 PetscCall(DMGetWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVec)); in DMPlexComputeResidualByKey()
5291 PetscCall(PetscArrayzero(elemVec, numCells * totDim)); in DMPlexComputeResidualByKey()
5296 PetscCall(DMGetWorkArray(dm, numFaces * totDim, MPIU_SCALAR, &fluxL)); in DMPlexComputeResidualByKey()
5297 PetscCall(DMGetWorkArray(dm, numFaces * totDim, MPIU_SCALAR, &fluxR)); in DMPlexComputeResidualByKey()
5298 PetscCall(PetscArrayzero(fluxL, numFaces * totDim)); in DMPlexComputeResidualByKey()
5299 PetscCall(PetscArrayzero(fluxR, numFaces * totDim)); in DMPlexComputeResidualByKey()
5336 …t * totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a… in DMPlexComputeResidualByKey()
5354 … if (mesh->printFEM > 1) PetscCall(DMPrintCellVector(cell, name, totDim, &elemVec[cind * totDim])); in DMPlexComputeResidualByKey()
5361 … PetscCall(DMPlexVecSetClosure(dm, section, locF, cell, &elemVec[cind * totDim], ADD_ALL_VALUES)); in DMPlexComputeResidualByKey()
5399 … PetscCall(DMPrintCellVector(face, "Residual: left state", pdim, &uL[iface * totDim + foff])); in DMPlexComputeResidualByKey()
5400 … PetscCall(DMPrintCellVector(face, "Residual: right state", pdim, &uR[iface * totDim + foff])); in DMPlexComputeResidualByKey()
5401 … PetscCall(DMPrintCellVector(face, "Residual: left flux", pdim, &fluxL[iface * totDim + foff])); in DMPlexComputeResidualByKey()
5402 … PetscCall(DMPrintCellVector(face, "Residual: right flux", pdim, &fluxR[iface * totDim + foff])); in DMPlexComputeResidualByKey()
5405 if (fL) fL[d] -= fluxL[iface * totDim + foff + d]; in DMPlexComputeResidualByKey()
5406 if (fR) fR[d] += fluxR[iface * totDim + foff + d]; in DMPlexComputeResidualByKey()
5450 PetscCall(DMRestoreWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVec)); in DMPlexComputeResidualByKey()
5455 PetscCall(DMRestoreWorkArray(dm, numFaces * totDim, MPIU_SCALAR, &fluxL)); in DMPlexComputeResidualByKey()
5456 PetscCall(DMRestoreWorkArray(dm, numFaces * totDim, MPIU_SCALAR, &fluxR)); in DMPlexComputeResidualByKey()
5578 …PetscInt Nf, f, totDim, totDimIn, totDimAux[3], totDimScale[3], numChunks, cellChunkSize, c… in DMPlexComputeResidualHybridByKey() local
5603 PetscCall(PetscDSGetTotalDimension(ds, &totDim)); in DMPlexComputeResidualHybridByKey()
5713 PetscCall(DMGetWorkArray(dm, cellChunkSize * totDim, MPIU_SCALAR, &elemVecNeg)); in DMPlexComputeResidualHybridByKey()
5714 PetscCall(DMGetWorkArray(dm, cellChunkSize * totDim, MPIU_SCALAR, &elemVecPos)); in DMPlexComputeResidualHybridByKey()
5715 PetscCall(DMGetWorkArray(dm, cellChunkSize * totDim, MPIU_SCALAR, &elemVecCoh)); in DMPlexComputeResidualHybridByKey()
5719 PetscCall(PetscArrayzero(elemVecNeg, cellChunkSize * totDim)); in DMPlexComputeResidualHybridByKey()
5720 PetscCall(PetscArrayzero(elemVecPos, cellChunkSize * totDim)); in DMPlexComputeResidualHybridByKey()
5721 PetscCall(PetscArrayzero(elemVecCoh, cellChunkSize * totDim)); in DMPlexComputeResidualHybridByKey()
5786 …Aux[0], PetscSafePointerPlusOffset(a[0], offset * totDimAux[0]), t, &elemVecNeg[offset * totDim])); in DMPlexComputeResidualHybridByKey()
5788 …Aux[1], PetscSafePointerPlusOffset(a[1], offset * totDimAux[1]), t, &elemVecPos[offset * totDim])); in DMPlexComputeResidualHybridByKey()
5790 …Aux[2], PetscSafePointerPlusOffset(a[2], offset * totDimAux[2]), t, &elemVecCoh[offset * totDim])); in DMPlexComputeResidualHybridByKey()
5804 PetscInt Nb, off = cind * totDim, soff = cind * totDimScale[0]; in DMPlexComputeResidualHybridByKey()
5823 …for (i = cind * totDim; i < (cind + 1) * totDim; ++i) elemVecCoh[i] += elemVecNeg[i] + elemVecPos[… in DMPlexComputeResidualHybridByKey()
5825 … (mesh->printFEM > 1) PetscCall(DMPrintCellVector(cell, name, totDim, &elemVecCoh[cind * totDim])); in DMPlexComputeResidualHybridByKey()
5832 …PetscCall(DMPlexVecSetClosure(dm, section, locF, cell, &elemVecCoh[cind * totDim], ADD_ALL_VALUES)… in DMPlexComputeResidualHybridByKey()
5838 PetscCall(DMRestoreWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVecNeg)); in DMPlexComputeResidualHybridByKey()
5839 PetscCall(DMRestoreWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVecPos)); in DMPlexComputeResidualHybridByKey()
5840 PetscCall(DMRestoreWorkArray(dm, numCells * totDim, MPIU_SCALAR, &elemVecCoh)); in DMPlexComputeResidualHybridByKey()
5917 PetscInt Nf, totDim, totDimAux = 0; in DMPlexComputeBdJacobianSingleByLabel() local
5927 PetscCall(PetscDSGetTotalDimension(ds, &totDim)); in DMPlexComputeBdJacobianSingleByLabel()
5969 …totDim, &u, (locX_t ? (size_t)numFaces * totDim : 0), &u_t, (hasJac ? (size_t)numFaces * totDim * … in DMPlexComputeBdJacobianSingleByLabel()
5988 for (i = 0; i < totDim; ++i) u[face * totDim + i] = x[i]; in DMPlexComputeBdJacobianSingleByLabel()
5992 for (i = 0; i < totDim; ++i) u_t[face * totDim + i] = x[i]; in DMPlexComputeBdJacobianSingleByLabel()
6003 if (elemMat) PetscCall(PetscArrayzero(elemMat, numFaces * totDim * totDim)); in DMPlexComputeBdJacobianSingleByLabel()
6004 if (elemMatP) PetscCall(PetscArrayzero(elemMatP, numFaces * totDim * totDim)); in DMPlexComputeBdJacobianSingleByLabel()
6034 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a, of… in DMPlexComputeBdJacobianSingleByLabel()
6036 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a, of… in DMPlexComputeBdJacobianSingleByLabel()
6045 …ointTensor_Internal(dm, tdm, tv, support[0], PETSC_TRUE, totDim, &elemMat[face * totDim * totDim])… in DMPlexComputeBdJacobianSingleByLabel()
6046 …intTensor_Internal(dm, tdm, tv, support[0], PETSC_TRUE, totDim, &elemMatP[face * totDim * totDim])… in DMPlexComputeBdJacobianSingleByLabel()
6049 … 1) PetscCall(DMPrintCellMatrix(point, "BdJacobian", totDim, totDim, &elemMat[face * totDim * totD… in DMPlexComputeBdJacobianSingleByLabel()
6050 … globalSection, mesh->useMatClPerm, Jac, support[0], &elemMat[face * totDim * totDim], ADD_VALUES)… in DMPlexComputeBdJacobianSingleByLabel()
6052 …1) PetscCall(DMPrintCellMatrix(point, "BdJacobian", totDim, totDim, &elemMatP[face * totDim * totD… in DMPlexComputeBdJacobianSingleByLabel()
6053 …lobalSection, mesh->useMatClPerm, JacP, support[0], &elemMatP[face * totDim * totDim], ADD_VALUES)… in DMPlexComputeBdJacobianSingleByLabel()
6056 … 1) PetscCall(DMPrintCellMatrix(point, "BdJacobian", totDim, totDim, &elemMat[face * totDim * totD… in DMPlexComputeBdJacobianSingleByLabel()
6057 … globalSection, mesh->useMatClPerm, Jac, support[0], &elemMat[face * totDim * totDim], ADD_VALUES)… in DMPlexComputeBdJacobianSingleByLabel()
6185 PetscInt totDim, totDimAux = 0, cStart, cEnd, numCells, c; in DMPlexComputeJacobianByKey() local
6211 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeJacobianByKey()
6219 …totDim, &u, (locX_t ? (size_t)numCells * totDim : 0), &u_t, (hasJac ? (size_t)numCells * totDim * … in DMPlexComputeJacobianByKey()
6228 for (i = 0; i < totDim; ++i) u[cind * totDim + i] = x[i]; in DMPlexComputeJacobianByKey()
6232 for (i = 0; i < totDim; ++i) u_t[cind * totDim + i] = x_t[i]; in DMPlexComputeJacobianByKey()
6243 if (hasJac) PetscCall(PetscArrayzero(elemMat, numCells * totDim * totDim)); in DMPlexComputeJacobianByKey()
6244 if (hasPrec) PetscCall(PetscArrayzero(elemMatP, numCells * totDim * totDim)); in DMPlexComputeJacobianByKey()
6245 if (hasDyn) PetscCall(PetscArrayzero(elemMatD, numCells * totDim * totDim)); in DMPlexComputeJacobianByKey()
6287 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, PetscSafePointerPlusOffset(a, … in DMPlexComputeJacobianByKey()
6291 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, PetscSafePointerPlusOffset(a, … in DMPlexComputeJacobianByKey()
6295 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, PetscSafePointerPlusOffset(a, … in DMPlexComputeJacobianByKey()
6305 for (c = 0; c < numCells * totDim * totDim; ++c) elemMat[c] += X_tShift * elemMatD[c]; in DMPlexComputeJacobianByKey()
6321 const PetscInt eOffset = cind * totDim * totDim; in DMPlexComputeJacobianByKey()
6329 if (hasJac) elemMat[eOffset + i * totDim + i] = vol; in DMPlexComputeJacobianByKey()
6330 elemMatP[eOffset + i * totDim + i] = vol; in DMPlexComputeJacobianByKey()
6332 elemMat[eOffset + i * totDim + i] = vol; in DMPlexComputeJacobianByKey()
6347 …sformPointTensor_Internal(dm, tdm, tv, cell, PETSC_TRUE, totDim, &elemMat[cind * totDim * totDim])… in DMPlexComputeJacobianByKey()
6350 …rintFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMat[cind * totDim * totD… in DMPlexComputeJacobianByKey()
6351 …ction, globalSection, mesh->useMatClPerm, Jac, cell, &elemMat[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianByKey()
6353 …intFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMatP[cind * totDim * totD… in DMPlexComputeJacobianByKey()
6354 …ion, globalSection, mesh->useMatClPerm, JacP, cell, &elemMatP[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianByKey()
6357 …rintFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMat[cind * totDim * totD… in DMPlexComputeJacobianByKey()
6358 …tion, globalSection, mesh->useMatClPerm, JacP, cell, &elemMat[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianByKey()
6400 PetscInt totDim, ctotDim, totDimAux = 0, cStart, cEnd, numCells; in DMPlexComputeJacobianByKeyGeneral() local
6433 PetscCall(PetscDSGetTotalDimension(rds, &totDim)); in DMPlexComputeJacobianByKeyGeneral()
6442 …totDim, &u, (locX_t ? (size_t)numCells * totDim : 0), &u_t, (hasJac ? (size_t)numCells * totDim * … in DMPlexComputeJacobianByKeyGeneral()
6451 for (i = 0; i < totDim; ++i) u[cind * totDim + i] = x[i]; in DMPlexComputeJacobianByKeyGeneral()
6455 for (i = 0; i < totDim; ++i) u_t[cind * totDim + i] = x_t[i]; in DMPlexComputeJacobianByKeyGeneral()
6466 if (hasJac) PetscCall(PetscArrayzero(elemMat, numCells * totDim * ctotDim)); in DMPlexComputeJacobianByKeyGeneral()
6467 if (hasPrec) PetscCall(PetscArrayzero(elemMatP, numCells * totDim * ctotDim)); in DMPlexComputeJacobianByKeyGeneral()
6468 if (hasDyn) PetscCall(PetscArrayzero(elemMatD, numCells * totDim * ctotDim)); in DMPlexComputeJacobianByKeyGeneral()
6510 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a, of… in DMPlexComputeJacobianByKeyGeneral()
6514 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a, of… in DMPlexComputeJacobianByKeyGeneral()
6518 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), dsAux, PetscSafePointerPlusOffset(a, of… in DMPlexComputeJacobianByKeyGeneral()
6528 for (PetscInt c = 0; c < numCells * totDim * ctotDim; ++c) elemMat[c] += X_tShift * elemMatD[c]; in DMPlexComputeJacobianByKeyGeneral()
6544 const PetscInt eOffset = cind * totDim * ctotDim; in DMPlexComputeJacobianByKeyGeneral()
6570 …ransformPointTensor_Internal(dmr, tdm, tv, cell, PETSC_TRUE, totDim, &elemMat[cind * totDim * ctot… in DMPlexComputeJacobianByKeyGeneral()
6573 …f (printFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, ctotDim, &elemMat[cind * totDim *… in DMPlexComputeJacobianByKeyGeneral()
6574 …lPerm, dmc, csection, cglobalSection, clPerm, Jac, cell, &elemMat[cind * totDim * ctotDim], ADD_VA… in DMPlexComputeJacobianByKeyGeneral()
6576 … (printFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, ctotDim, &elemMatP[cind * totDim *… in DMPlexComputeJacobianByKeyGeneral()
6577 …erm, dmc, csection, cglobalSection, clPerm, JacP, cell, &elemMatP[cind * totDim * ctotDim], ADD_VA… in DMPlexComputeJacobianByKeyGeneral()
6580 …f (printFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, ctotDim, &elemMat[cind * totDim *… in DMPlexComputeJacobianByKeyGeneral()
6581 …Perm, dmc, csection, cglobalSection, clPerm, JacP, cell, &elemMat[cind * totDim * ctotDim], ADD_VA… in DMPlexComputeJacobianByKeyGeneral()
6659 …PetscInt Nf, fieldI, fieldJ, totDim, totDimIn, totDimAux[3], totDimScale[3], numChunks, cel… in DMPlexComputeJacobianHybridByKey() local
6684 PetscCall(PetscDSGetTotalDimension(ds, &totDim)); in DMPlexComputeJacobianHybridByKey()
6784 …PetscCall(DMGetWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMat… in DMPlexComputeJacobianHybridByKey()
6785 …PetscCall(DMGetWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMat… in DMPlexComputeJacobianHybridByKey()
6786 …PetscCall(DMGetWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMat… in DMPlexComputeJacobianHybridByKey()
6787 …PetscCall(DMGetWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMa… in DMPlexComputeJacobianHybridByKey()
6788 …PetscCall(DMGetWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMa… in DMPlexComputeJacobianHybridByKey()
6789 …PetscCall(DMGetWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &elemMa… in DMPlexComputeJacobianHybridByKey()
6794 PetscCall(PetscArrayzero(elemMatNeg, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6795 PetscCall(PetscArrayzero(elemMatPos, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6796 PetscCall(PetscArrayzero(elemMatCoh, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6799 PetscCall(PetscArrayzero(elemMatNegP, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6800 PetscCall(PetscArrayzero(elemMatPosP, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6801 PetscCall(PetscArrayzero(elemMatCohP, cellChunkSize * totDim * totDim)); in DMPlexComputeJacobianHybridByKey()
6871 …interPlusOffset(a[0], offset * totDimAux[0]), t, X_tShift, &elemMatNeg[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6873 …interPlusOffset(a[1], offset * totDimAux[1]), t, X_tShift, &elemMatPos[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6877 …In), dsAux[0], &a[0][offset * totDimAux[0]], t, X_tShift, &elemMatNegP[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6879 …In), dsAux[1], &a[1][offset * totDimAux[1]], t, X_tShift, &elemMatPosP[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6883 …interPlusOffset(a[2], offset * totDimAux[2]), t, X_tShift, &elemMatCoh[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6887 …In), dsAux[2], &a[2][offset * totDimAux[2]], t, X_tShift, &elemMatCohP[offset * totDim * totDim])); in DMPlexComputeJacobianHybridByKey()
6898 const PetscInt cind = c - cS, coff = cind * totDim * totDim; in DMPlexComputeJacobianHybridByKey()
6913 …totDim; ++j) elemMatCoh[coff + (off + i) * totDim + j] += s[0][soff + i] * elemMatNeg[coff + (off … in DMPlexComputeJacobianHybridByKey()
6915 …totDim; ++j) elemMatCohP[coff + (off + i) * totDim + j] += s[0][soff + i] * elemMatNegP[coff + (of… in DMPlexComputeJacobianHybridByKey()
6922 …0; j < totDim; ++j) elemMatCoh[coff + (off + i) * totDim + j] += elemMatNeg[coff + (off + i) * tot… in DMPlexComputeJacobianHybridByKey()
6924 …; j < totDim; ++j) elemMatCohP[coff + (off + i) * totDim + j] += elemMatNegP[coff + (off + i) * to… in DMPlexComputeJacobianHybridByKey()
6930 …for (i = 0; i < totDim * totDim; ++i) elemMatCoh[coff + i] += elemMatNeg[coff + i] + elemMatPos[co… in DMPlexComputeJacobianHybridByKey()
6932 …for (i = 0; i < totDim * totDim; ++i) elemMatCohP[coff + i] += elemMatNegP[coff + i] + elemMatPosP… in DMPlexComputeJacobianHybridByKey()
6936 …ntFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMatCoh[cind * totDim * tot… in DMPlexComputeJacobianHybridByKey()
6937 …on, globalSection, mesh->useMatClPerm, Jac, cell, &elemMatCoh[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianHybridByKey()
6939 …ntFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMatCohP[cind * totDim * to… in DMPlexComputeJacobianHybridByKey()
6940 …losure(plex, section, globalSection, JacP, cell, &elemMatCohP[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianHybridByKey()
6942 …ntFEM > 1) PetscCall(DMPrintCellMatrix(cell, name, totDim, totDim, &elemMatCoh[cind * totDim * tot… in DMPlexComputeJacobianHybridByKey()
6943 …n, globalSection, mesh->useMatClPerm, JacP, cell, &elemMatCoh[cind * totDim * totDim], ADD_VALUES)… in DMPlexComputeJacobianHybridByKey()
6949 …PetscCall(DMRestoreWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &ele… in DMPlexComputeJacobianHybridByKey()
6950 …PetscCall(DMRestoreWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &ele… in DMPlexComputeJacobianHybridByKey()
6951 …PetscCall(DMRestoreWorkArray(dm, hasBdJac ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &ele… in DMPlexComputeJacobianHybridByKey()
6952 …PetscCall(DMRestoreWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &el… in DMPlexComputeJacobianHybridByKey()
6953 …PetscCall(DMRestoreWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &el… in DMPlexComputeJacobianHybridByKey()
6954 …PetscCall(DMRestoreWorkArray(dm, hasBdPrec ? cellChunkSize * totDim * totDim : 0, MPIU_SCALAR, &el… in DMPlexComputeJacobianHybridByKey()
7016 PetscInt totDim, totDimAux = 0, cStart, cEnd, numCells, c; in DMPlexComputeJacobianActionByKey() local
7028 PetscCall(PetscDSGetTotalDimension(prob, &totDim)); in DMPlexComputeJacobianActionByKey()
7041 …totDim, &u, (locX_t ? (size_t)numCells * totDim : 0), &u_t, numCells * totDim * totDim, &elemMat, … in DMPlexComputeJacobianActionByKey()
7051 for (i = 0; i < totDim; ++i) u[cind * totDim + i] = x[i]; in DMPlexComputeJacobianActionByKey()
7055 for (i = 0; i < totDim; ++i) u_t[cind * totDim + i] = x_t[i]; in DMPlexComputeJacobianActionByKey()
7066 for (i = 0; i < totDim; ++i) y[cind * totDim + i] = x[i]; in DMPlexComputeJacobianActionByKey()
7069 PetscCall(PetscArrayzero(elemMat, numCells * totDim * totDim)); in DMPlexComputeJacobianActionByKey()
7070 if (hasDyn) PetscCall(PetscArrayzero(elemMatD, numCells * totDim * totDim)); in DMPlexComputeJacobianActionByKey()
7106 …totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, PetscSafePointerPlusOffset(a, … in DMPlexComputeJacobianActionByKey()
7109 …fset * totDim], PetscSafePointerPlusOffset(u_t, offset * totDim), probAux, &a[offset * totDimAux],… in DMPlexComputeJacobianActionByKey()
7118 for (c = 0; c < numCells * totDim * totDim; ++c) elemMat[c] += X_tShift * elemMatD[c]; in DMPlexComputeJacobianActionByKey()
7127 PetscCall(PetscBLASIntCast(totDim, &M)); in DMPlexComputeJacobianActionByKey()
7128 …S("BLASgemv", BLASgemv_("N", &M, &M, &a, &elemMat[cind * totDim * totDim], &M, &y[cind * totDim], … in DMPlexComputeJacobianActionByKey()
7130 PetscCall(DMPrintCellMatrix(c, name, totDim, totDim, &elemMat[cind * totDim * totDim])); in DMPlexComputeJacobianActionByKey()
7131 PetscCall(DMPrintCellVector(c, "Y", totDim, &y[cind * totDim])); in DMPlexComputeJacobianActionByKey()
7132 PetscCall(DMPrintCellVector(c, "Z", totDim, z)); in DMPlexComputeJacobianActionByKey()