Allocates and manages CUDA device memory.
Definition at line 129 of file infer_cuda_utils.h.
Public Member Functions | |
CudaDeviceMem (size_t size, int gpuId=0) | |
~CudaDeviceMem () override | |
void | grow (size_t bytes) override |
size_t | bytes () const |
template<typename T > | |
T * | ptr () const |
void * | ptr () const |
int | devId () const |
InferMemType | type () const |
void | reuse () |
Protected Member Functions | |
void | move_copy (SysMem &&o) |
DISABLE_CLASS_COPY (SysMem) | |
Protected Attributes | |
void * | m_Buf = nullptr |
size_t | m_Size = 0 |
int | m_DevId = 0 |
InferMemType | m_Type = InferMemType::kNone |
nvdsinferserver::CudaDeviceMem::CudaDeviceMem | ( | size_t | size, |
int | gpuId = 0 |
||
) |
|
override |
|
inlineinherited |
Definition at line 90 of file infer_cuda_utils.h.
References nvdsinferserver::SysMem::m_Size.
|
inlineinherited |
Definition at line 98 of file infer_cuda_utils.h.
References nvdsinferserver::SysMem::m_DevId.
|
protectedinherited |
|
overridevirtual |
Implements nvdsinferserver::SysMem.
|
inlineprotectedinherited |
Definition at line 107 of file infer_cuda_utils.h.
References nvdsinferserver::kNone, nvdsinferserver::SysMem::m_Buf, nvdsinferserver::SysMem::m_DevId, nvdsinferserver::SysMem::m_Size, and nvdsinferserver::SysMem::m_Type.
|
inlineinherited |
Definition at line 93 of file infer_cuda_utils.h.
References nvdsinferserver::SysMem::m_Buf.
|
inlineinherited |
Definition at line 97 of file infer_cuda_utils.h.
References nvdsinferserver::SysMem::m_Buf.
|
inlineinherited |
Definition at line 100 of file infer_cuda_utils.h.
|
inlineinherited |
Definition at line 99 of file infer_cuda_utils.h.
References nvdsinferserver::SysMem::m_Type.
|
protectedinherited |
Definition at line 120 of file infer_cuda_utils.h.
Referenced by nvdsinferserver::SysMem::move_copy(), and nvdsinferserver::SysMem::ptr().
|
protectedinherited |
Definition at line 122 of file infer_cuda_utils.h.
Referenced by nvdsinferserver::SysMem::devId(), and nvdsinferserver::SysMem::move_copy().
|
protectedinherited |
Definition at line 121 of file infer_cuda_utils.h.
Referenced by nvdsinferserver::SysMem::bytes(), and nvdsinferserver::SysMem::move_copy().
|
protectedinherited |
Definition at line 123 of file infer_cuda_utils.h.
Referenced by nvdsinferserver::SysMem::move_copy(), and nvdsinferserver::SysMem::type().