Skip to content

GitLab

Commit 960c22af authored by Peter Eastman's avatar Peter Eastman
Browse files

Minor bug fix

parent a711ce2a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
platforms/cuda/src/CudaKernels.cpp
View file @ 960c22af
...@@ -48,16 +48,13 @@ void CudaCalcForcesAndEnergyKernel::beginComputation(ContextImpl& context, bool ...@@ -48,16 +48,13 @@ void CudaCalcForcesAndEnergyKernel::beginComputation(ContextImpl& context, bool
_gpuContext* gpu = data.gpu; _gpuContext* gpu = data.gpu;
if (data.nonbondedMethod != NO_CUTOFF && data.computeForceCount%100 == 0) if (data.nonbondedMethod != NO_CUTOFF && data.computeForceCount%100 == 0)
gpuReorderAtoms(gpu); gpuReorderAtoms(gpu);
if (includeForces) data.computeForceCount++;
data.computeForceCount++;
if (gpu->bIncludeGBSA || gpu->bIncludeGBVI) if (gpu->bIncludeGBSA || gpu->bIncludeGBVI)
kClearBornSumAndForces(gpu); kClearBornSumAndForces(gpu);
else if (includeForces) else if (includeForces)
kClearForces(gpu); kClearForces(gpu);
if (includeEnergy) { if (includeEnergy)
data.stepCount++;
kClearEnergy(gpu); kClearEnergy(gpu);
}
} }
double CudaCalcForcesAndEnergyKernel::finishComputation(ContextImpl& context, bool includeForces, bool includeEnergy) { double CudaCalcForcesAndEnergyKernel::finishComputation(ContextImpl& context, bool includeForces, bool includeEnergy) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment