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

55
tests/operators/prepared_hdiv_mass.cpp
View File

@@ -30,6 +30,61 @@ namespace {
}
} // namespace
TEST_CASE(
"Prepared Hdiv Geometry Variation Preserves Reference Piola Contractions",
tags::gravity_prepared_jacobian_accuracy
) {
auto args = test_utils::setup_args();
fem::FEM f = fem::setup_fem(args.mesh_file, args, 0);
REQUIRE(f.okay());
operators::PreparedMappedHDivMassOperator preparedOperator(f, *f.domainMapperStateless);
const MPI_Comm communicator = f.gravityFluxFes->GetComm();
const mfem::Vector first = prepared_test::make_deterministic_vector(f.gravityFluxFes->GetTrueVSize(), 0.31);
const mfem::Vector second = prepared_test::make_deterministic_vector(f.gravityFluxFes->GetTrueVSize(), 0.79);
const mfem::Vector direction = prepared_test::make_displacement(f, 0.43);
mfem::Vector firstAction;
bool hasStellar = false;
bool hasVacuum = false;
for (int element = 0; element < f.mesh->GetNE(); ++element) {
const bool vacuum = f.domainMapperStateless->IsCompactifiedElement(*f.mesh->GetElementTransformation(element));
hasVacuum = hasVacuum || vacuum;
hasStellar = hasStellar || !vacuum;
}
const int localDomains[2]{hasStellar ? 1 : 0, hasVacuum ? 1 : 0};
int globalDomains[2]{};
REQUIRE(MPI_Allreduce(localDomains, globalDomains, 2, MPI_INT, MPI_MAX, communicator) == MPI_SUCCESS);
REQUIRE(globalDomains[0] != 0);
REQUIRE(globalDomains[1] != 0);
// The stateless path still constructs each physically mapped RT basis;
// compare both an undeformed and a changed prepared geometry, including
// Kelvin exterior elements, against the compact forward/dual contractions.
for (const double scale : {0.0, 0.7}) {
const mfem::Vector displacementTrue = prepared_test::make_displacement(f, scale);
const mfem::Vector displacement = preparedOperator.GetDisplacementMap().gather(displacementTrue);
preparedOperator.Prepare(displacement);
preparedOperator.MultDisplacementVariationTrue(first, direction, firstAction);
mfem::Vector referenceAction;
operators::kernels::apply_mapped_hdiv_mass_variation(
f, *f.domainMapperStateless, first, displacementTrue, direction, referenceAction
);
const double error = prepared_test::relative_error(firstAction, referenceAction, communicator);
INFO("Deformation scale = " << scale);
INFO("Prepared/stateless geometry-variation relative error = " << error);
CHECK(error < 2.0e-11);
}
mfem::Vector secondAction;
preparedOperator.MultDisplacementVariationTrue(second, direction, secondAction);
const double firstSecond = prepared_test::global_dot(first, secondAction, communicator);
const double secondFirst = prepared_test::global_dot(second, firstAction, communicator);
CHECK(prepared_test::relative_scalar_error(firstSecond, secondFirst) < 2.0e-11);
CHECK(preparedOperator.GetPreparationCount() == 2);
}
TEST_CASE(
"Prepared Mapped Hdiv Mass Reports Invalid Candidate Geometry Without Unwinding",
tags::gravity_prepared_unit &tags::geometry