mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2025-02-11 16:55:23 +00:00
cf5ef5635f
Introducing CUDA buffer pool with generic CUDA memory support. Likewise GL memory, any elements which are able to access CUDA device memory directly can map this CUDA memory without upload/download overhead via the "GST_MAP_CUDA" map flag. Also usual GstMemory map/unmap is also possible with internal staging memory. For staging, CUDA Host allocated memory is used (see CuMemAllocHost API). The memory is allowing system access but has lower overhead during GPU upload/download than normal system memory. Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/1633>
125 lines
2.9 KiB
C
125 lines
2.9 KiB
C
/* CUDA stub header
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifndef __GST_CUDA_STUB_H__
|
|
#define __GST_CUDA_STUB_H__
|
|
|
|
#include <glib.h>
|
|
|
|
G_BEGIN_DECLS
|
|
|
|
typedef gpointer CUcontext;
|
|
typedef gpointer CUgraphicsResource;
|
|
typedef gpointer CUstream;
|
|
typedef gpointer CUarray;
|
|
|
|
typedef guintptr CUdeviceptr;
|
|
typedef gint CUdevice;
|
|
|
|
typedef enum
|
|
{
|
|
CUDA_SUCCESS = 0,
|
|
} CUresult;
|
|
|
|
typedef enum
|
|
{
|
|
CU_MEMORYTYPE_HOST = 1,
|
|
CU_MEMORYTYPE_DEVICE = 2,
|
|
CU_MEMORYTYPE_ARRAY = 3,
|
|
CU_MEMORYTYPE_UNIFIED = 4,
|
|
} CUmemorytype;
|
|
|
|
typedef enum
|
|
{
|
|
CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR = 75,
|
|
CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR = 76,
|
|
} CUdevice_attribute;
|
|
|
|
typedef enum
|
|
{
|
|
CU_GRAPHICS_REGISTER_FLAGS_NONE = 0x00,
|
|
CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY = 0x01,
|
|
CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD = 0x02,
|
|
} CUgraphicsRegisterFlags;
|
|
|
|
typedef enum
|
|
{
|
|
CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE = 0x00,
|
|
CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY = 0x01,
|
|
CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD = 0x02,
|
|
} CUgraphicsMapResourceFlags;
|
|
|
|
typedef enum
|
|
{
|
|
CU_STREAM_DEFAULT = 0x0,
|
|
CU_STREAM_NON_BLOCKING = 0x1
|
|
} CUstream_flags;
|
|
|
|
typedef struct
|
|
{
|
|
gsize srcXInBytes;
|
|
gsize srcY;
|
|
CUmemorytype srcMemoryType;
|
|
gconstpointer srcHost;
|
|
CUdeviceptr srcDevice;
|
|
CUarray srcArray;
|
|
gsize srcPitch;
|
|
|
|
gsize dstXInBytes;
|
|
gsize dstY;
|
|
CUmemorytype dstMemoryType;
|
|
gpointer dstHost;
|
|
CUdeviceptr dstDevice;
|
|
CUarray dstArray;
|
|
gsize dstPitch;
|
|
|
|
gsize WidthInBytes;
|
|
gsize Height;
|
|
} CUDA_MEMCPY2D;
|
|
|
|
typedef enum
|
|
{
|
|
CU_GL_DEVICE_LIST_ALL = 0x01,
|
|
} CUGLDeviceList;
|
|
|
|
#define CUDA_VERSION 10000
|
|
|
|
#ifdef _WIN32
|
|
#define CUDAAPI __stdcall
|
|
#else
|
|
#define CUDAAPI
|
|
#endif
|
|
|
|
#define cuCtxCreate cuCtxCreate_v2
|
|
#define cuCtxDestroy cuCtxDestroy_v2
|
|
#define cuCtxPopCurrent cuCtxPopCurrent_v2
|
|
#define cuCtxPushCurrent cuCtxPushCurrent_v2
|
|
#define cuGraphicsResourceGetMappedPointer cuGraphicsResourceGetMappedPointer_v2
|
|
#define cuGraphicsResourceSetMapFlags cuGraphicsResourceSetMapFlags_v2
|
|
|
|
#define cuMemAlloc cuMemAlloc_v2
|
|
#define cuMemAllocPitch cuMemAllocPitch_v2
|
|
#define cuMemAllocHost cuMemAllocHost_v2
|
|
#define cuMemcpy2D cuMemcpy2D_v2
|
|
#define cuMemcpy2DAsync cuMemcpy2DAsync_v2
|
|
#define cuMemFree cuMemFree_v2
|
|
#define cuGLGetDevices cuGLGetDevices_v2
|
|
|
|
G_END_DECLS
|
|
|
|
#endif /* __GST_CUDA_STUB_H__ */
|