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
f03d9ae2
Commit
f03d9ae2
authored
Oct 30, 2018
by
Peter Eastman
Browse files
Use separate events for different stream synchronizations
parent
049db77b
Changes
2
Hide 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 @
f03d9ae2
...
@@ -701,7 +701,7 @@ private:
...
@@ -701,7 +701,7 @@ private:
Kernel
cpuPme
;
Kernel
cpuPme
;
PmeIO
*
pmeio
;
PmeIO
*
pmeio
;
CUstream
pmeStream
;
CUstream
pmeStream
;
CUevent
pmeSyncEvent
;
CUevent
pmeSyncEvent
,
paramsSyncEvent
;
CudaFFT3D
*
fft
;
CudaFFT3D
*
fft
;
cufftHandle
fftForward
;
cufftHandle
fftForward
;
cufftHandle
fftBackward
;
cufftHandle
fftBackward
;
...
...
platforms/cuda/src/CudaKernels.cpp
View file @
f03d9ae2
...
@@ -1561,6 +1561,7 @@ CudaCalcNonbondedForceKernel::~CudaCalcNonbondedForceKernel() {
...
@@ -1561,6 +1561,7 @@ CudaCalcNonbondedForceKernel::~CudaCalcNonbondedForceKernel() {
if (usePmeStream) {
if (usePmeStream) {
cuStreamDestroy(pmeStream);
cuStreamDestroy(pmeStream);
cuEventDestroy(pmeSyncEvent);
cuEventDestroy(pmeSyncEvent);
cuEventDestroy(paramsSyncEvent);
}
}
}
}
}
}
...
@@ -1862,6 +1863,7 @@ void CudaCalcNonbondedForceKernel::initialize(const System& system, const Nonbon
...
@@ -1862,6 +1863,7 @@ void CudaCalcNonbondedForceKernel::initialize(const System& system, const Nonbon
cufftSetStream(dispersionFftBackward, pmeStream);
cufftSetStream(dispersionFftBackward, pmeStream);
}
}
CHECK_RESULT(cuEventCreate(&pmeSyncEvent, CU_EVENT_DISABLE_TIMING), "Error creating event for NonbondedForce");
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();
int recipForceGroup = force.getReciprocalSpaceForceGroup();
if (recipForceGroup < 0)
if (recipForceGroup < 0)
recipForceGroup = force.getForceGroup();
recipForceGroup = force.getForceGroup();
...
@@ -2110,8 +2112,8 @@ double CudaCalcNonbondedForceKernel::execute(ContextImpl& context, bool includeF
...
@@ -2110,8 +2112,8 @@ double CudaCalcNonbondedForceKernel::execute(ContextImpl& context, bool includeF
}
}
cu.executeKernel(computeParamsKernel, ¶msArgs[0], cu.getPaddedNumAtoms());
cu.executeKernel(computeParamsKernel, ¶msArgs[0], cu.getPaddedNumAtoms());
if (usePmeStream) {
if (usePmeStream) {
cuEventRecord(p
me
SyncEvent, cu.getCurrentStream());
cuEventRecord(p
arams
SyncEvent, cu.getCurrentStream());
cuStreamWaitEvent(pmeStream, p
me
SyncEvent, 0);
cuStreamWaitEvent(pmeStream, p
arams
SyncEvent, 0);
}
}
energy = 0.0; // The Ewald self energy was computed in the kernel.
energy = 0.0; // The Ewald self energy was computed in the kernel.
recomputeParams = false;
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