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 32 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 37 of file gstnvdspreprocess_allocator.h.
void* GstNvDsPreProcessMemory::dev_memory_ptr |
Pointer to the memory allocated for the batch of frames (DGPU).
Definition at line 41 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 39 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 43 of file gstnvdspreprocess_allocator.h.
NvBufSurface* GstNvDsPreProcessMemory::surf |
surface corresponding to memory allocated
Definition at line 35 of file gstnvdspreprocess_allocator.h.