Spaces:
Runtime error
Runtime error
File size: 1,444 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 |
#include <unittest/unittest.h>
#include <thrust/sort.h>
#include <thrust/functional.h>
template <typename T, unsigned int N>
void _TestStableSortWithLargeKeys(void)
{
size_t n = (128 * 1024) / sizeof(FixedVector<T,N>);
thrust::host_vector< FixedVector<T,N> > h_keys(n);
for(size_t i = 0; i < n; i++)
// XXX Use proper random number generation facility.
h_keys[i] = FixedVector<T,N>(rand());
thrust::device_vector< FixedVector<T,N> > d_keys = h_keys;
thrust::stable_sort(h_keys.begin(), h_keys.end());
thrust::stable_sort(d_keys.begin(), d_keys.end());
ASSERT_EQUAL_QUIET(h_keys, d_keys);
}
void TestStableSortWithLargeKeys(void)
{
_TestStableSortWithLargeKeys<int, 1>();
_TestStableSortWithLargeKeys<int, 2>();
_TestStableSortWithLargeKeys<int, 4>();
_TestStableSortWithLargeKeys<int, 8>();
_TestStableSortWithLargeKeys<int, 16>();
_TestStableSortWithLargeKeys<int, 32>();
_TestStableSortWithLargeKeys<int, 64>();
_TestStableSortWithLargeKeys<int, 128>();
_TestStableSortWithLargeKeys<int, 256>();
// XXX these take too long to compile
// _TestStableSortWithLargeKeys<int, 512>();
// _TestStableSortWithLargeKeys<int, 1024>();
// _TestStableSortWithLargeKeys<int, 2048>();
// _TestStableSortWithLargeKeys<int, 4096>();
// _TestStableSortWithLargeKeys<int, 8192>();
}
DECLARE_UNITTEST(TestStableSortWithLargeKeys);
|