Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
tsoc
openmm
Commits
689e2bc9
Unverified
Commit
689e2bc9
authored
Oct 31, 2018
by
peastman
Committed by
GitHub
Oct 31, 2018
Browse files
Merge pull request #2195 from peastman/syncevent
Use separate events for different stream synchronizations
parents
049db77b
f03d9ae2
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
3 deletions
+5
-3
platforms/cuda/include/CudaKernels.h
platforms/cuda/include/CudaKernels.h
+1
-1
platforms/cuda/src/CudaKernels.cpp
platforms/cuda/src/CudaKernels.cpp
+4
-2
No files found.
platforms/cuda/include/CudaKernels.h
View file @
689e2bc9
...
...
@@ -701,7 +701,7 @@ private:
Kernel
cpuPme
;
PmeIO
*
pmeio
;
CUstream
pmeStream
;
CUevent
pmeSyncEvent
;
CUevent
pmeSyncEvent
,
paramsSyncEvent
;
CudaFFT3D
*
fft
;
cufftHandle
fftForward
;
cufftHandle
fftBackward
;
...
...
platforms/cuda/src/CudaKernels.cpp
View file @
689e2bc9
...
...
@@ -1561,6 +1561,7 @@ CudaCalcNonbondedForceKernel::~CudaCalcNonbondedForceKernel() {
if (usePmeStream) {
cuStreamDestroy(pmeStream);
cuEventDestroy(pmeSyncEvent);
cuEventDestroy(paramsSyncEvent);
}
}
}
...
...
@@ -1862,6 +1863,7 @@ void CudaCalcNonbondedForceKernel::initialize(const System& system, const Nonbon
cufftSetStream(dispersionFftBackward, pmeStream);
}
CHECK_RESULT(cuEventCreate(&pmeSyncEvent, CU_EVENT_DISABLE_TIMING), "Error creating event for NonbondedForce");
CHECK_RESULT(cuEventCreate(¶msSyncEvent, CU_EVENT_DISABLE_TIMING), "Error creating event for NonbondedForce");
int recipForceGroup = force.getReciprocalSpaceForceGroup();
if (recipForceGroup < 0)
recipForceGroup = force.getForceGroup();
...
...
@@ -2110,8 +2112,8 @@ double CudaCalcNonbondedForceKernel::execute(ContextImpl& context, bool includeF
}
cu.executeKernel(computeParamsKernel, ¶msArgs[0], cu.getPaddedNumAtoms());
if (usePmeStream) {
cuEventRecord(p
me
SyncEvent, cu.getCurrentStream());
cuStreamWaitEvent(pmeStream, p
me
SyncEvent, 0);
cuEventRecord(p
arams
SyncEvent, cu.getCurrentStream());
cuStreamWaitEvent(pmeStream, p
arams
SyncEvent, 0);
}
energy = 0.0; // The Ewald self energy was computed in the kernel.
recomputeParams = false;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment