NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 14:58:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor correction in unit test

Browse Source
This commit is contained in:
Yunsong Wang
2022-02-06 20:19:21 -05:00
parent e05bf002f7
commit 6159d9c6cb
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
testing/state.cu
View File

@@ -58,8 +58,9 @@ void test_streams()
state_tester state{bench};
// Test non-owning stream
state.set_cuda_stream(nvbench::cuda_stream{cudaStreamDefault, false});
ASSERT(state.get_cuda_stream() == cudaStreamDefault);
cudaStream_t default_stream = 0;
state.set_cuda_stream(nvbench::cuda_stream{default_stream, false});
ASSERT(state.get_cuda_stream() == default_stream);
// Test owning stream
auto stream = nvbench::cuda_stream{};