23 lines
432 B
Plaintext
23 lines
432 B
Plaintext
|
#include <thrust/execution_policy.h>
|
||
|
|
#include <thrust/system/cuda/detail/util.h>
|
||
|
|
|
||
|
|
#include <thread>
|
||
|
|
|
||
|
|
#include <unittest/unittest.h>
|
||
|
|
|
||
|
|
void verify_stream()
|
||
|
|
{
|
||
|
|
auto exec = thrust::device;
|
||
|
|
auto stream = thrust::cuda_cub::stream(exec);
|
||
|
|
ASSERT_EQUAL(stream, cudaStreamPerThread);
|
||
|
|
}
|
||
|
|
|
||
|
|
void TestPerThreadDefaultStream()
|
||
|
|
{
|
||
|
|
verify_stream();
|
||
|
|
|
||
|
|
std::thread t(verify_stream);
|
||
|
|
t.join();
|
||
|
|
}
|
||
|
|
DECLARE_UNITTEST(TestPerThreadDefaultStream);
|