Spaces:
Runtime error
Runtime error
File size: 2,340 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 |
#include <thrust/mr/allocator.h>
#include <thrust/mr/new.h>
#include <thrust/mr/pool.h>
#include <thrust/mr/disjoint_pool.h>
#include <thrust/device_vector.h>
#include <thrust/device_ptr.h>
#include <cassert>
template<typename Vec>
void do_stuff_with_vector(typename Vec::allocator_type alloc)
{
Vec v1(alloc);
v1.push_back(1);
assert(v1.back() == 1);
Vec v2(alloc);
v2 = v1;
v1.swap(v2);
v1.clear();
v1.resize(2);
assert(v1.size() == 2);
}
int main()
{
thrust::mr::new_delete_resource memres;
{
// no virtual calls will be issued
typedef thrust::mr::allocator<int, thrust::mr::new_delete_resource> Alloc;
Alloc alloc(&memres);
do_stuff_with_vector<thrust::host_vector<int, Alloc> >(alloc);
}
{
// virtual calls will be issued - wrapping in a polymorphic wrapper
thrust::mr::polymorphic_adaptor_resource<void *> adaptor(&memres);
typedef thrust::mr::polymorphic_allocator<int, void *> Alloc;
Alloc alloc(&adaptor);
do_stuff_with_vector<thrust::host_vector<int, Alloc> >(alloc);
}
{
// use the global device_ptr-flavored device memory resource
typedef thrust::device_ptr_memory_resource<thrust::device_memory_resource> Resource;
thrust::mr::polymorphic_adaptor_resource<thrust::device_ptr<void> > adaptor(
thrust::mr::get_global_resource<Resource>()
);
typedef thrust::mr::polymorphic_allocator<int, thrust::device_ptr<void> > Alloc;
Alloc alloc(&adaptor);
do_stuff_with_vector<thrust::device_vector<int, Alloc> >(alloc);
}
typedef thrust::mr::unsynchronized_pool_resource<
thrust::mr::new_delete_resource
> Pool;
Pool pool(&memres);
{
typedef thrust::mr::allocator<int, Pool> Alloc;
Alloc alloc(&pool);
do_stuff_with_vector<thrust::host_vector<int, Alloc> >(alloc);
}
typedef thrust::mr::disjoint_unsynchronized_pool_resource<
thrust::mr::new_delete_resource,
thrust::mr::new_delete_resource
> DisjointPool;
DisjointPool disjoint_pool(&memres, &memres);
{
typedef thrust::mr::allocator<int, DisjointPool> Alloc;
Alloc alloc(&disjoint_pool);
do_stuff_with_vector<thrust::host_vector<int, Alloc> >(alloc);
}
}
|