Gromacs  2026.0-dev-20241213-9ac17bb
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
List of all members | Public Member Functions
gmx::NNPotForceProvider Class Referencefinal

#include <gromacs/applied_forces/nnpot/nnpotforceprovider.h>

+ Inheritance diagram for gmx::NNPotForceProvider:
+ Collaboration diagram for gmx::NNPotForceProvider:

Description

NNPotForceProvider class

Implements the IForceProvider interface for the NNPot force provider.

Public Member Functions

 NNPotForceProvider (const NNPotParameters &, const MDLogger *logger)
 
 ~NNPotForceProvider ()
 Destroy force provider for NNPot.
 
void calculateForces (const ForceProviderInput &fInput, ForceProviderOutput *fOutput) override
 Calculate forces of NNPot. More...
 
void gatherAtomNumbersIndices ()
 Gather atom numbers and indices. Triggered on AtomsRedistributed signal.
 
void gatherAtomPositions (ArrayRef< const RVec > pos)
 Gather atom positions for NN input.
 

Member Function Documentation

void gmx::NNPotForceProvider::calculateForces ( const ForceProviderInput fInput,
ForceProviderOutput fOutput 
)
overridevirtual

Calculate forces of NNPot.

Prepares the input for the neural network model triggers model inference.

Parameters
[in]fInputinput for force provider
[out]fOutputoutput for force provider

Implements gmx::IForceProvider.


The documentation for this class was generated from the following files: