Skip to content
Snippets Groups Projects
Commit 2221f9c0 authored by Hkorb's avatar Hkorb
Browse files

simplifiy streamIsRegistered check in StreamManager

parent ddc3d0f1
1 merge request!170Kernel templetization and efficiency improvements
...@@ -52,7 +52,7 @@ void CudaStreamManager::terminateStreams() ...@@ -52,7 +52,7 @@ void CudaStreamManager::terminateStreams()
bool CudaStreamManager::streamIsRegistered(StreamIndex streamIndex) bool CudaStreamManager::streamIsRegistered(StreamIndex streamIndex)
{ {
return cudaStreams.find(streamIndex) != cudaStreams.end(); return cudaStreams.count(streamIndex) == 1;
} }
cudaStream_t &CudaStreamManager::getStream(StreamIndex streamIndex) cudaStream_t &CudaStreamManager::getStream(StreamIndex streamIndex)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment