Lines Matching refs:chunkSize
2768 PetscInt numChunks, chunkSize, chunk, Nr, offset; in DMPlexComputeBdIntegral_Internal() local
2813 chunkSize = numBatches * batchSize; in DMPlexComputeBdIntegral_Internal()
2815 numChunks = numFaces / chunkSize; in DMPlexComputeBdIntegral_Internal()
2816 Nr = numFaces % chunkSize; in DMPlexComputeBdIntegral_Internal()
2821 … PetscCall(PetscFEGeomGetChunk(fgeom, chunk * chunkSize, (chunk + 1) * chunkSize, &chunkGeom)); in DMPlexComputeBdIntegral_Internal()
2822 …chunkSize, chunkGeom, &u[chunk * chunkSize * totDim], probAux, PetscSafePointerPlusOffset(a, chunk… in DMPlexComputeBdIntegral_Internal()
4634 …PetscInt Nf, fieldI, fieldJ, maxDegree, numCells, cStart, cEnd, numChunks, chunkSize, chunk… in DMPlexComputeJacobian_Patch_Internal() local
4683 chunkSize = numBatches * batchSize; in DMPlexComputeJacobian_Patch_Internal()
4684 numChunks = numCells / chunkSize + numCells % chunkSize; in DMPlexComputeJacobian_Patch_Internal()
4687 chunkSize = numCells; in DMPlexComputeJacobian_Patch_Internal()
4691 …tDimAux + ((hasJac ? 1 : 0) + (hasPrec ? 1 : 0) + (hasDyn ? 1 : 0)) * totDim * totDim) * chunkSize; 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()
4697 a = dmAux ? (sz = chunkSize * totDimAux, 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()
4720 for (chunk = 0; chunk < numChunks; ++chunk, offCell += chunkSize) { in DMPlexComputeJacobian_Patch_Internal()
4721 const PetscInt Ncell = PetscMin(chunkSize, numCells - offCell); in DMPlexComputeJacobian_Patch_Internal()
4763 for (c = 0; c < chunkSize; ++c, eOffset += totDim * totDim) { in DMPlexComputeJacobian_Patch_Internal()
4774 for (c = 0; c < chunkSize * totDim * totDim; ++c) elemMat[c] += X_tShift * elemMatD[c]; in DMPlexComputeJacobian_Patch_Internal()
4792 …c ? 1 : 0) + (hasPrec ? 1 : 0) + (hasDyn ? 1 : 0)) * totDim * totDim) * chunkSize, MPIU_SCALAR, &w… in DMPlexComputeJacobian_Patch_Internal()