1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- #ifndef BOOST_COMPUTE_ALLOCATOR_PINNED_ALLOCATOR_HPP
- #define BOOST_COMPUTE_ALLOCATOR_PINNED_ALLOCATOR_HPP
- #include <boost/compute/allocator/buffer_allocator.hpp>
- namespace boost {
- namespace compute {
- template<class T>
- class pinned_allocator : public buffer_allocator<T>
- {
- public:
- explicit pinned_allocator(const context &context)
- : buffer_allocator<T>(context)
- {
- buffer_allocator<T>::set_mem_flags(
- buffer::read_write | buffer::alloc_host_ptr
- );
- }
- pinned_allocator(const pinned_allocator<T> &other)
- : buffer_allocator<T>(other)
- {
- }
- pinned_allocator<T>& operator=(const pinned_allocator<T> &other)
- {
- if(this != &other){
- buffer_allocator<T>::operator=(other);
- }
- return *this;
- }
- ~pinned_allocator()
- {
- }
- };
- }
- }
- #endif
|