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

Add nvbench::state stream tests

Browse Source
This commit is contained in:
Yunsong Wang
2022-02-04 16:55:29 -05:00
parent 439ffec1c8
commit 470beda9f0
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
testing/state.cu
View File

@@ -51,6 +51,23 @@ struct state_tester : public nvbench::state
using nvbench::detail::state_tester;
void test_streams()
{
dummy_bench bench;
state_tester state{bench};
// Test non-owning stream
state.set_cuda_stream(nvbench::cuda_stream{cudaStreamDefault, false});
ASSERT(state.get_cuda_stream() == cudaStreamDefault);
// Test owning stream
auto stream = nvbench::cuda_stream{};
auto gold = stream.get_stream();
state.set_cuda_stream(std::move(stream));
ASSERT(state.get_cuda_stream() == gold);
}
void test_params()
{
dummy_bench bench;
@@ -110,6 +127,7 @@ void test_defaults()
int main()
{
test_streams();
test_params();
test_summaries();
test_defaults();