Memory resource that wraps direct allocations on a CUDA device. More...
#include <vecmem/memory/cuda/device_memory_resource.hpp>
Inheritance diagram for vecmem::cuda::device_memory_resource:
Public Member Functions | |
| VECMEM_CUDA_EXPORT | device_memory_resource (int device=INVALID_DEVICE) |
| Construct a CUDA device resource for a specific device. | |
| VECMEM_CUDA_EXPORT | ~device_memory_resource () |
| Destructor. | |
Static Public Attributes | |
| static constexpr int | INVALID_DEVICE = -1 |
| Invalid/default device identifier. | |
Private Member Functions | |
Function(s) implementing @c vecmem::memory_resource | |
| virtual VECMEM_CUDA_EXPORT void * | do_allocate (std::size_t, std::size_t) override final |
| Allocate memory on the selected device. | |
| virtual VECMEM_CUDA_EXPORT void | do_deallocate (void *p, std::size_t, std::size_t) override final |
| De-allocate a previously allocated memory block on the selected device. | |
| virtual VECMEM_CUDA_EXPORT bool | do_is_equal (const memory_resource &other) const noexcept override final |
Compares *this for equality with other. | |
Detailed Description
Memory resource that wraps direct allocations on a CUDA device.
This is an allocator-type memory resource (that is to say, it only allocates, it does not try to manage memory in a smart way) that works for CUDA device memory. Each instance is bound to a specific device.
Constructor & Destructor Documentation
◆ device_memory_resource()
| vecmem::cuda::device_memory_resource::device_memory_resource | ( | int | device = INVALID_DEVICE | ) |
Construct a CUDA device resource for a specific device.
This constructor takes a device identifier argument which works in the same way as in standard CUDA code. If the device number is positive, that device is selected. If the device number is negative, the currently selected device is used.
- Note
- The default device is resolved at resource construction time, not at allocation time.
- Parameters
-
[in] device CUDA identifier for the device to use
The documentation for this class was generated from the following files:
- cuda/include/vecmem/memory/cuda/device_memory_resource.hpp
- cuda/src/memory/device_memory_resource.cpp
