Gromacs
2026.0-dev-20241213-9ac17bb
|
#include <gromacs/applied_forces/nnpot/nnpotforceprovider.h>
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. | |
|
overridevirtual |
Calculate forces of NNPot.
Prepares the input for the neural network model triggers model inference.
[in] | fInput | input for force provider |
[out] | fOutput | output for force provider |
Implements gmx::IForceProvider.