tboudreaux/MeanField
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(allocations): reduced overall allocations by 95%, increaseed jacobian applicatin by 2x

Browse Source 
This commit uses global pre allocated work space to dramatically reduce memory usage and allocation time
This commit is contained in:
Emily Boudreaux
2026-09-10 06:50:56 -04:00
parent b3c04d507a
commit 75cc638739
66 changed files with 207183 additions and 99552 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
libmeanfield/impl/operators/prepared_pressure_force.cpp
View File

@@ -492,9 +492,6 @@ namespace mean_field::operators {
m_elements.reserve(m_fem.mesh->GetNE());
mfem::Vector enthalpyShape;
mfem::DenseMatrix displacementDShape;
for (int elementId = 0; elementId < m_fem.mesh->GetNE(); ++elementId) {
mfem::ElementTransformation *transformation = m_fem.mesh->GetElementTransformation(elementId);
@@ -559,29 +556,11 @@ namespace mean_field::operators {
"an unexpected vector DOF count."
);
data.enthalpyBasis.SetSize(quadraturePointCount, enthalpyDofCount);
data.referenceTestGradients.resize(quadraturePointCount);
const fem::ReferenceTableCache &referenceTables = m_fem.GetReferenceTables();
data.enthalpyReferenceTable = referenceTables.GetScalarTable(enthalpyElement, *data.integrationRule);
data.displacementReferenceTable =
referenceTables.GetScalarTable(displacementElement, *data.integrationRule);
data.physicalTestGradients.resize(quadraturePointCount);
enthalpyShape.SetSize(enthalpyDofCount);
displacementDShape.SetSize(scalarDisplacementDofCount, dimension);
for (int quadraturePoint = 0; quadraturePoint < quadraturePointCount; ++quadraturePoint) {
const mfem::IntegrationPoint &integrationPoint = data.integrationRule->IntPoint(quadraturePoint);
enthalpyElement.CalcShape(integrationPoint, enthalpyShape);
displacementElement.CalcDShape(integrationPoint, displacementDShape);
for (int enthalpyDof = 0; enthalpyDof < enthalpyDofCount; ++enthalpyDof) {
data.enthalpyBasis(quadraturePoint, enthalpyDof) = enthalpyShape(enthalpyDof);
}
data.referenceTestGradients[quadraturePoint] = displacementDShape;
}
}
}
@@ -591,6 +570,7 @@ namespace mean_field::operators {
true_to_local(*m_fem.displacementFes, m_baseDisplacementTrue, displacementLocal);
mapping::DomainMapper::Workspace workspace(m_fem.mesh->Dimension());
mapping::VolumeMappingContext mappingContext;
mfem::Vector elementDisplacement;
mfem::Vector elementCompactification;
@@ -634,14 +614,15 @@ namespace mean_field::operators {
const int quadraturePointCount = data.integrationRule->GetNPoints();
MFEM_VERIFY(
static_cast<int>(data.referenceTestGradients.size()) == quadraturePointCount,
data.displacementReferenceTable != nullptr &&
data.displacementReferenceTable->GetPointCount() == quadraturePointCount,
"Prepared pressure-force geometry has inconsistent "
"static gradient data."
);
data.quadratureWeights.SetSize(quadraturePointCount);
data.baseMappingContexts.resize(quadraturePointCount);
data.baseMappingContexts.SetSize(quadraturePointCount, m_fem.mesh->Dimension());
data.physicalTestGradients.resize(quadraturePointCount);
@@ -650,9 +631,7 @@ namespace mean_field::operators {
transformation->SetIntPoint(&integrationPoint);
mapping::VolumeMappingContext &mappingContext = data.baseMappingContexts[quadraturePoint];
const mapping::MappingStatus mappingStatus = m_domainMapper.EvaluateVolume(
const mapping::MappingStatus mappingStatus = m_domainMapper.EvaluateVolume(
mappingData, *transformation, integrationPoint, workspace, mappingContext
);
@@ -669,11 +648,13 @@ namespace mean_field::operators {
return mapping_rejection(mapping::MappingStatus::non_positive_determinant);
}
data.quadratureWeights(quadraturePoint) = quadratureWeight;
data.baseMappingContexts.Store(quadraturePoint, mappingContext);
data.quadratureWeights(quadraturePoint) = quadratureWeight;
const mfem::DenseMatrix &referenceTestGradient = data.referenceTestGradients[quadraturePoint];
const mfem::DenseMatrix &referenceTestGradient =
data.displacementReferenceTable->GetGradients(quadraturePoint);
mfem::DenseMatrix &physicalTestGradient = data.physicalTestGradients[quadraturePoint];
mfem::DenseMatrix &physicalTestGradient = data.physicalTestGradients[quadraturePoint];
MFEM_VERIFY(
referenceTestGradient.Width() == mappingContext.quadrature.J_inv.Height() &&
@@ -714,9 +695,11 @@ namespace mean_field::operators {
data.enthalpyDofTransformation->InvTransformPrimal(elementEnthalpy);
}
const int quadraturePointCount = data.enthalpyBasis.Height();
const mfem::DenseMatrix &enthalpyBasis = data.GetEnthalpyBasis();
const int enthalpyDofCount = data.enthalpyBasis.Width();
const int quadraturePointCount = enthalpyBasis.Height();
const int enthalpyDofCount = enthalpyBasis.Width();
const mfem::FiniteElement &displacementElement = *m_fem.displacementFes->GetFE(data.elementId);
@@ -734,7 +717,7 @@ namespace mean_field::operators {
quadratureEnthalpy.SetSize(quadraturePointCount);
data.enthalpyBasis.Mult(elementEnthalpy, quadratureEnthalpy);
enthalpyBasis.Mult(elementEnthalpy, quadratureEnthalpy);
data.pressure.SetSize(quadraturePointCount);
@@ -814,8 +797,8 @@ namespace mean_field::operators {
data.elementResidual(vectorDof) -= residualContribution;
for (int enthalpyDof = 0; enthalpyDof < enthalpyDofCount; ++enthalpyDof) {
const double jacobianContribution = pressureDerivative * weightedTestGradient *
data.enthalpyBasis(quadraturePoint, enthalpyDof);
const double jacobianContribution =
pressureDerivative * weightedTestGradient * enthalpyBasis(quadraturePoint, enthalpyDof);
if (!std::isfinite(jacobianContribution)) {
materialFailure = non_finite_rejection();
continue;
@@ -841,8 +824,10 @@ namespace mean_field::operators {
MFEM_VERIFY(
data.baseDisplacementData.has_value() && data.compactificationData.has_value() &&
static_cast<int>(data.baseMappingContexts.size()) == quadraturePointCount &&
static_cast<int>(data.referenceTestGradients.size()) == quadraturePointCount &&
data.baseMappingContexts.GetPointCount() == quadraturePointCount &&
data.baseMappingContexts.GetDimension() == dimension &&
data.displacementReferenceTable != nullptr &&
data.displacementReferenceTable->GetPointCount() == quadraturePointCount &&
static_cast<int>(data.physicalTestGradients.size()) == quadraturePointCount &&
data.quadratureWeights.Size() == quadraturePointCount &&
data.pressure.Size() == quadraturePointCount,
@@ -852,7 +837,7 @@ namespace mean_field::operators {
for (int quadraturePoint = 0; quadraturePoint < quadraturePointCount; ++quadraturePoint) {
MFEM_VERIFY(
data.referenceTestGradients[quadraturePoint].Width() == dimension &&
data.displacementReferenceTable->GetGradients(quadraturePoint).Width() == dimension &&
data.physicalTestGradients[quadraturePoint].Width() == dimension,
"Prepared pressure-force displacement Jacobian has "
"a gradient with the wrong dimension."
@@ -978,6 +963,7 @@ namespace mean_field::operators {
mfem::Vector elementAction;
mfem::DenseMatrix referenceDisplacementJacobian;
mfem::DenseMatrix inverseElementJacobian;
mfem::DenseMatrix inverseElementJacobianVariation;
mfem::DenseMatrix matrixTemporary;
mfem::DenseMatrix physicalTestGradientVariation;
@@ -1016,7 +1002,7 @@ namespace mean_field::operators {
);
MFEM_VERIFY(
static_cast<int>(data.baseMappingContexts.size()) == quadraturePointCount &&
data.baseMappingContexts.GetPointCount() == quadraturePointCount &&
data.pressure.Size() == quadraturePointCount,
"Prepared pressure-force displacement Jacobian has "
"stale quadrature data."
@@ -1032,12 +1018,12 @@ namespace mean_field::operators {
physicalTestGradientVariation.SetSize(scalarDisplacementDofCount, dimension);
for (int quadraturePoint = 0; quadraturePoint < quadraturePointCount; ++quadraturePoint) {
const mfem::DenseMatrix &referenceTestGradient = data.referenceTestGradients[quadraturePoint];
const mfem::DenseMatrix &referenceTestGradient =
data.displacementReferenceTable->GetGradients(quadraturePoint);
mfem::MultAtB(directionDofs, referenceTestGradient, referenceDisplacementJacobian);
const mfem::DenseMatrix &inverseElementJacobian =
data.baseMappingContexts[quadraturePoint].quadrature.J_inv;
data.baseMappingContexts.LoadInverseJacobian(quadraturePoint, inverseElementJacobian);
mfem::Mult(inverseElementJacobian, referenceDisplacementJacobian, matrixTemporary);
double logarithmicJacobianVariation{0.0};