This file describes the custom memory allocator for the Gstreamer TensorRT plugin.
The allocator allocates memory for a specified batch_size of frames of resolution equal to the network input resolution and RGBA color format. The frames are allocated on device memory. Holds the pointer for the allocated memory.
Definition at line 42 of file gstnvdspreprocess_allocator.h.
Data Fields | |
NvBufSurface * | surf |
surface corresponding to memory allocated More... | |
std::vector< CUgraphicsResource > | cuda_resources |
Vector of cuda resources created by registering the above egl images in CUDA. More... | |
std::vector< CUeglFrame > | egl_frames |
Vector of CUDA eglFrames created by mapping the above cuda resources. More... | |
void * | dev_memory_ptr |
Pointer to the memory allocated for the batch of frames (DGPU). More... | |
std::vector< void * > | frame_memory_ptrs |
Vector of pointer to individual frame memories in the batch memory. More... | |
std::vector<CUgraphicsResource> GstNvDsPreProcessMemory::cuda_resources |
Vector of cuda resources created by registering the above egl images in CUDA.
Definition at line 47 of file gstnvdspreprocess_allocator.h.
void* GstNvDsPreProcessMemory::dev_memory_ptr |
Pointer to the memory allocated for the batch of frames (DGPU).
Definition at line 51 of file gstnvdspreprocess_allocator.h.
std::vector<CUeglFrame> GstNvDsPreProcessMemory::egl_frames |
Vector of CUDA eglFrames created by mapping the above cuda resources.
Definition at line 49 of file gstnvdspreprocess_allocator.h.
std::vector<void *> GstNvDsPreProcessMemory::frame_memory_ptrs |
Vector of pointer to individual frame memories in the batch memory.
Definition at line 53 of file gstnvdspreprocess_allocator.h.
NvBufSurface* GstNvDsPreProcessMemory::surf |
surface corresponding to memory allocated
Definition at line 45 of file gstnvdspreprocess_allocator.h.