Spaces:
Runtime error
Runtime error
File size: 3,032 Bytes
be11144 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
#include <unittest/unittest.h>
#include <thrust/mismatch.h>
#include <thrust/execution_policy.h>
template<typename ExecutionPolicy, typename Iterator1, typename Iterator2, typename Iterator3>
__global__ void mismatch_kernel(ExecutionPolicy exec, Iterator1 first1, Iterator1 last1, Iterator2 first2, Iterator3 result)
{
*result = thrust::mismatch(exec, first1, last1, first2);
}
template<typename ExecutionPolicy>
void TestMismatchDevice(ExecutionPolicy exec)
{
thrust::device_vector<int> a(4);
thrust::device_vector<int> b(4);
a[0] = 1; b[0] = 1;
a[1] = 2; b[1] = 2;
a[2] = 3; b[2] = 4;
a[3] = 4; b[3] = 3;
typedef thrust::pair<
typename thrust::device_vector<int>::iterator,
typename thrust::device_vector<int>::iterator
> pair_type;
thrust::device_vector<pair_type> d_result(1);
mismatch_kernel<<<1,1>>>(exec, a.begin(), a.end(), b.begin(), d_result.begin());
{
cudaError_t const err = cudaDeviceSynchronize();
ASSERT_EQUAL(cudaSuccess, err);
}
ASSERT_EQUAL(2, ((pair_type)d_result[0]).first - a.begin());
ASSERT_EQUAL(2, ((pair_type)d_result[0]).second - b.begin());
b[2] = 3;
mismatch_kernel<<<1,1>>>(exec, a.begin(), a.end(), b.begin(), d_result.begin());
{
cudaError_t const err = cudaDeviceSynchronize();
ASSERT_EQUAL(cudaSuccess, err);
}
ASSERT_EQUAL(3, ((pair_type)d_result[0]).first - a.begin());
ASSERT_EQUAL(3, ((pair_type)d_result[0]).second - b.begin());
b[3] = 4;
mismatch_kernel<<<1,1>>>(exec, a.begin(), a.end(), b.begin(), d_result.begin());
{
cudaError_t const err = cudaDeviceSynchronize();
ASSERT_EQUAL(cudaSuccess, err);
}
ASSERT_EQUAL(4, ((pair_type)d_result[0]).first - a.begin());
ASSERT_EQUAL(4, ((pair_type)d_result[0]).second - b.begin());
}
void TestMismatchDeviceSeq()
{
TestMismatchDevice(thrust::seq);
}
DECLARE_UNITTEST(TestMismatchDeviceSeq);
void TestMismatchDeviceDevice()
{
TestMismatchDevice(thrust::device);
}
DECLARE_UNITTEST(TestMismatchDeviceDevice);
void TestMismatchCudaStreams()
{
typedef thrust::device_vector<int> Vector;
Vector a(4); Vector b(4);
a[0] = 1; b[0] = 1;
a[1] = 2; b[1] = 2;
a[2] = 3; b[2] = 4;
a[3] = 4; b[3] = 3;
cudaStream_t s;
cudaStreamCreate(&s);
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).first - a.begin(), 2);
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).second - b.begin(), 2);
b[2] = 3;
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).first - a.begin(), 3);
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).second - b.begin(), 3);
b[3] = 4;
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).first - a.begin(), 4);
ASSERT_EQUAL(thrust::mismatch(thrust::cuda::par.on(s), a.begin(), a.end(), b.begin()).second - b.begin(), 4);
cudaStreamDestroy(s);
}
DECLARE_UNITTEST(TestMismatchCudaStreams);
|