Merge pull request #2413 from peastman/context

Make sure correct CUDA context is current

Merge pull request #2413 from peastman/context
Make sure correct CUDA context is current
509fa77e · peastman · GitHub · 5e8265c7 · 2f6a3569 · 509fa77e
Unverified Commit 509fa77e authored Oct 01, 2019 by peastman Committed by GitHub Oct 01, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

platforms/cuda/src/CudaKernels.cpp platforms/cuda/src/CudaKernels.cpp +1 -0

No files found.
--- a/platforms/cuda/src/CudaKernels.cpp
+++ b/platforms/cuda/src/CudaKernels.cpp
@@ -120,6 +120,7 @@ void CudaCalcForcesAndEnergyKernel::beginComputation(ContextImpl& context, bool
 }
 double CudaCalcForcesAndEnergyKernel::finishComputation(ContextImpl& context, bool includeForces, bool includeEnergy, int groups, bool& valid) {
+    cu.setAsCurrent();
    cu.getBondedUtilities().computeInteractions(groups);
    cu.getNonbondedUtilities().computeInteractions(groups, includeForces, includeEnergy);
    double sum = 0.0;