1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#include <unittest/unittest.h>
#include <thrust/detail/config.h>
#include <thrust/detail/caching_allocator.h>
template<typename Allocator>
void test_implementation(Allocator alloc)
{
typedef typename thrust::detail::allocator_traits<Allocator> Traits;
typedef typename Allocator::pointer Ptr;
Ptr p = Traits::allocate(alloc, 123);
Traits::deallocate(alloc, p, 123);
Ptr p2 = Traits::allocate(alloc, 123);
ASSERT_EQUAL(p, p2);
}
void TestSingleDeviceTLSCachingAllocator()
{
test_implementation(thrust::detail::single_device_tls_caching_allocator());
};
DECLARE_UNITTEST(TestSingleDeviceTLSCachingAllocator);
|
