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

perf(jacobian-action): major updates to jacobian action application by removing redudant quadrature work. ~5x increase in speed

Browse Source
This commit is contained in:
Emily Boudreaux
2026-09-02 17:01:50 -04:00
parent 85500fef3b
commit 25510008dd
74 changed files with 8967 additions and 814 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
libmeanfield/interface/operators/prepared_hdiv_mass.cppm
View File

@@ -2,6 +2,7 @@ module;
#include <cstdint>
#include <memory>
#include <mfem.hpp>
#include <vector>
export module mean_field:operators.prepared_hdiv_mass;
export import :fem;
@@ -21,6 +22,11 @@ export namespace mean_field::operators {
const mfem::Vector &gravity_gradient,
mfem::Vector &action
) const override;
void MultDisplacementVariationTrue(
const mfem::Vector &gravityGradientTrue,
const mfem::Vector &displacementVariationTrue,
mfem::Vector &actionVariationTrue
) const;
void AssembleDiagonal(mfem::Vector &diagonal) const override;
void AssembleTrueDiagonal(mfem::Vector &diagonal) const;
@@ -31,6 +37,21 @@ export namespace mean_field::operators {
[[nodiscard]] const field::FieldDofMap &GetDisplacementMap() const noexcept;
private:
struct ElementVariationData {
int elementId{-1};
mfem::Array<int> gravityGradientDofs;
mfem::Array<int> displacementDofs;
mfem::Array<int> compactificationDofs;
mfem::DofTransformation *gravityGradientDofTransformation{nullptr};
mfem::DofTransformation *displacementDofTransformation{nullptr};
mfem::Vector baseDisplacement;
mfem::Vector compactification;
const mfem::IntegrationRule *integrationRule{nullptr};
mfem::DenseMatrix frozenMappingData;
};
void PrepareVariationData();
const fem::FEM &m_fem;
const mapping::DomainMapper &m_domain_mapper;
@@ -48,6 +69,21 @@ export namespace mean_field::operators {
mutable mfem::Vector m_action_true;
mutable mfem::Vector m_domain_action_true;
mfem::Vector m_displacement_true;
std::vector<ElementVariationData> m_variationElements;
mutable mapping::DomainMapper::Workspace m_variationWorkspace;
mutable mapping::VolumeMappingContext m_baseMappingContext;
mutable mapping::VolumeMappingVariation m_mappingVariation;
mutable mfem::Vector m_gravityGradientLocal;
mutable mfem::Vector m_displacementVariationLocal;
mutable mfem::Vector m_localVariationAction;
mutable mfem::Vector m_elementGravityGradient;
mutable mfem::Vector m_elementDisplacementVariation;
mutable mfem::Vector m_elementVariationAction;
mutable mfem::Vector m_gravityGradientValue;
mutable mfem::Vector m_massTensorVariationAction;
mutable mfem::DenseMatrix m_gravityGradientShape;
mutable mfem::DenseMatrix m_massTensorVariation;
std::uint64_t m_preparation_count{0};
bool m_is_prepared{false};
};