Gromacs
2024.3
|
#include "config.h"
#include <string>
#include "gromacs/gpu_utils/gpu_utils.h"
#include "gromacs/math/vec.h"
#include "gromacs/mdtypes/simulation_workload.h"
#include "gromacs/nbnxm/nbnxm.h"
#include "gromacs/pbcutil/ishift.h"
#include "gromacs/timing/gpu_timing.h"
#include "gromacs/timing/wallcycle.h"
#include "gromacs/utility/stringutil.h"
#include "gpu_common_utils.h"
#include "nbnxm_gpu.h"
Common functions for the different NBNXN GPU implementations.
Functions | |
static void | Nbnxm::countPruneKernelTime (Nbnxm::GpuTimers *timers, gmx_wallclock_gpu_nbnxn_t *timings, const InteractionLocality iloc) |
Count pruning kernel time if either kernel has been triggered. More... | |
static void | Nbnxm::gpu_reduce_staged_outputs (const NBStagingData &nbst, const InteractionLocality iLocality, const bool reduceEnergies, const bool reduceFshift, real *e_lj, real *e_el, rvec *fshift) |
Reduce data staged internally in the nbnxn module. More... | |
template<typename GpuPairlist > | |
static void | Nbnxm::gpu_accumulate_timings (gmx_wallclock_gpu_nbnxn_t *timings, Nbnxm::GpuTimers *timers, const GpuPairlist *plist, AtomLocality atomLocality, const gmx::StepWorkload &stepWork, bool doTiming) |
Do the per-step timing accounting of the nonbonded tasks. More... | |
bool | Nbnxm::gpu_try_finish_task (NbnxmGpu *nb, const gmx::StepWorkload &stepWork, const AtomLocality aloc, real *e_lj, real *e_el, gmx::ArrayRef< gmx::RVec > shiftForces, GpuTaskCompletion completionKind) |
Attempts to complete nonbonded GPU task. More... | |
float | Nbnxm::gpu_wait_finish_task (NbnxmGpu *nb, const gmx::StepWorkload &stepWork, AtomLocality aloc, real *e_lj, real *e_el, gmx::ArrayRef< gmx::RVec > shiftForces, gmx_wallcycle *wcycle) |
Wait for the asynchronously launched nonbonded tasks and data transfers to finish. More... | |