Spaces:
Runtime error
Runtime error
File size: 2,684 Bytes
28958dc |
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 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
#include <unittest/unittest.h>
#include <thrust/device_vector.h>
#include <thrust/device_ptr.h>
void TestDevicePointerManipulation(void)
{
thrust::device_vector<int> data(5);
thrust::device_ptr<int> begin(&data[0]);
thrust::device_ptr<int> end(&data[0] + 5);
ASSERT_EQUAL(end - begin, 5);
begin++;
begin--;
ASSERT_EQUAL(end - begin, 5);
begin += 1;
begin -= 1;
ASSERT_EQUAL(end - begin, 5);
begin = begin + (int) 1;
begin = begin - (int) 1;
ASSERT_EQUAL(end - begin, 5);
begin = begin + (unsigned int) 1;
begin = begin - (unsigned int) 1;
ASSERT_EQUAL(end - begin, 5);
begin = begin + (size_t) 1;
begin = begin - (size_t) 1;
ASSERT_EQUAL(end - begin, 5);
begin = begin + (ptrdiff_t) 1;
begin = begin - (ptrdiff_t) 1;
ASSERT_EQUAL(end - begin, 5);
begin = begin + (thrust::device_ptr<int>::difference_type) 1;
begin = begin - (thrust::device_ptr<int>::difference_type) 1;
ASSERT_EQUAL(end - begin, 5);
}
DECLARE_UNITTEST(TestDevicePointerManipulation);
void TestMakeDevicePointer(void)
{
typedef int T;
T *raw_ptr = 0;
thrust::device_ptr<T> p0 = thrust::device_pointer_cast(raw_ptr);
ASSERT_EQUAL(thrust::raw_pointer_cast(p0), raw_ptr);
thrust::device_ptr<T> p1 = thrust::device_pointer_cast(p0);
ASSERT_EQUAL(p0, p1);
}
DECLARE_UNITTEST(TestMakeDevicePointer);
template<typename Vector>
void TestRawPointerCast(void)
{
typedef typename Vector::value_type T;
Vector vec(3);
T * first;
T * last;
first = thrust::raw_pointer_cast(&vec[0]);
last = thrust::raw_pointer_cast(&vec[3]);
ASSERT_EQUAL(last - first, 3);
first = thrust::raw_pointer_cast(&vec.front());
last = thrust::raw_pointer_cast(&vec.back());
ASSERT_EQUAL(last - first, 2);
// Do we want these to work?
//first = thrust::raw_pointer_cast(vec.begin());
//last = thrust::raw_pointer_cast(vec.end());
//ASSERT_EQUAL(last - first, 3);
}
DECLARE_VECTOR_UNITTEST(TestRawPointerCast);
#if THRUST_CPP_DIALECT >= 2011
template<typename T>
void TestDevicePointerNullptrCompatibility()
{
thrust::device_ptr<T> p0(nullptr);
ASSERT_EQUAL_QUIET(nullptr, p0);
ASSERT_EQUAL_QUIET(p0, nullptr);
p0 = nullptr;
ASSERT_EQUAL_QUIET(nullptr, p0);
ASSERT_EQUAL_QUIET(p0, nullptr);
}
DECLARE_GENERIC_UNITTEST(TestDevicePointerNullptrCompatibility);
template<typename T>
void TestDevicePointerBoolConversion()
{
thrust::device_ptr<T> p0(nullptr);
auto const b = bool(p0);
ASSERT_EQUAL_QUIET(false, b);
}
DECLARE_GENERIC_UNITTEST(TestDevicePointerBoolConversion);
#endif
|