mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-27 04:01:08 +00:00
174c9bfaa5
Required to support async memory allocation Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7427>
383 lines
13 KiB
C
383 lines
13 KiB
C
#pragma once
|
|
|
|
#include <gst/cuda/cuda-prelude.h>
|
|
#include <cuda.h>
|
|
#include <cudaGL.h>
|
|
|
|
#ifdef G_OS_WIN32
|
|
#ifndef INITGUID
|
|
#include <initguid.h>
|
|
#endif /* INITGUID */
|
|
|
|
#include <d3d11.h>
|
|
#include <dxgi.h>
|
|
#include <cudaD3D11.h>
|
|
#endif /* G_OS_WIN32 */
|
|
|
|
G_BEGIN_DECLS
|
|
|
|
/* cuda.h */
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuInit (unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGetErrorName (CUresult error,
|
|
const char **pStr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGetErrorString (CUresult error,
|
|
const char **pStr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxCreate (CUcontext * pctx,
|
|
unsigned int flags,
|
|
CUdevice dev);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxDestroy (CUcontext ctx);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxPopCurrent (CUcontext * pctx);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxPushCurrent (CUcontext ctx);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxSynchronize (void);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxEnablePeerAccess (CUcontext peerContext,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuCtxDisablePeerAccess (CUcontext peerContext);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsMapResources (unsigned int count,
|
|
CUgraphicsResource * resources,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsUnmapResources (unsigned int count,
|
|
CUgraphicsResource * resources,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsSubResourceGetMappedArray (CUarray * pArray,
|
|
CUgraphicsResource resource,
|
|
unsigned int arrayIndex,
|
|
unsigned int mipLevel);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsResourceGetMappedPointer (CUdeviceptr * pDevPtr,
|
|
size_t * pSize,
|
|
CUgraphicsResource resource);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsUnregisterResource (CUgraphicsResource resource);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAlloc (CUdeviceptr * dptr,
|
|
unsigned int bytesize);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAllocPitch (CUdeviceptr * dptr,
|
|
size_t * pPitch,
|
|
size_t WidthInBytes,
|
|
size_t Height,
|
|
unsigned int ElementSizeBytes);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAllocHost (void **pp,
|
|
unsigned int bytesize);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpy2D (const CUDA_MEMCPY2D * pCopy);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpy2DAsync (const CUDA_MEMCPY2D *pCopy, CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyDtoD (CUdeviceptr dstDevice,
|
|
CUdeviceptr srcDevice,
|
|
size_t ByteCount);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyDtoDAsync (CUdeviceptr dstDevice,
|
|
CUdeviceptr srcDevice,
|
|
size_t ByteCount,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyDtoH (void *dstHost,
|
|
CUdeviceptr srcDevice,
|
|
size_t ByteCount);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyDtoHAsync (void *dstHost,
|
|
CUdeviceptr srcDevice,
|
|
size_t ByteCount,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyHtoD (CUdeviceptr dstDevice,
|
|
const void *srcHost,
|
|
size_t ByteCount);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemcpyHtoDAsync (CUdeviceptr dstDevice,
|
|
const void *srcHost,
|
|
size_t ByteCount,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemFree (CUdeviceptr dptr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemFreeHost (void *p);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuStreamCreate (CUstream *phStream,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuStreamDestroy (CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuStreamSynchronize (CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDeviceGet (CUdevice * device,
|
|
int ordinal);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDeviceGetCount (int *count);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDeviceGetName (char *name,
|
|
int len,
|
|
CUdevice dev);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDeviceGetAttribute (int *pi,
|
|
CUdevice_attribute attrib,
|
|
CUdevice dev);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDeviceCanAccessPeer (int *canAccessPeer,
|
|
CUdevice dev,
|
|
CUdevice peerDev);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuDriverGetVersion (int * driverVersion);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuModuleLoadData (CUmodule* module,
|
|
const void *image);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuModuleUnload (CUmodule module);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuModuleGetFunction (CUfunction* hfunc,
|
|
CUmodule hmod,
|
|
const char* name);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuTexObjectCreate (CUtexObject *pTexObject,
|
|
const CUDA_RESOURCE_DESC *pResDesc,
|
|
const CUDA_TEXTURE_DESC *pTexDesc,
|
|
const CUDA_RESOURCE_VIEW_DESC *pResViewDesc);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuTexObjectDestroy (CUtexObject texObject);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuLaunchKernel (CUfunction f,
|
|
unsigned int gridDimX,
|
|
unsigned int gridDimY,
|
|
unsigned int gridDimZ,
|
|
unsigned int blockDimX,
|
|
unsigned int blockDimY,
|
|
unsigned int blockDimZ,
|
|
unsigned int sharedMemBytes,
|
|
CUstream hStream,
|
|
void **kernelParams,
|
|
void **extra);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuEventCreate (CUevent *phEvent,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuEventDestroy (CUevent hEvent);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuEventRecord (CUevent hEvent,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuEventSynchronize (CUevent hEvent);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuIpcGetEventHandle (CUipcEventHandle *pHandle,
|
|
CUevent event);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuIpcOpenEventHandle (CUevent* phEvent,
|
|
CUipcEventHandle handle);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuIpcGetMemHandle (CUipcMemHandle *pHandle,
|
|
CUdeviceptr dptr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuIpcOpenMemHandle (CUdeviceptr *pdptr,
|
|
CUipcMemHandle handle,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuIpcCloseMemHandle (CUdeviceptr dptr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAddressReserve (CUdeviceptr *ptr,
|
|
size_t size,
|
|
size_t alignment,
|
|
CUdeviceptr addr,
|
|
unsigned long long flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAddressFree (CUdeviceptr ptr,
|
|
size_t size);
|
|
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemCreate (CUmemGenericAllocationHandle *handle,
|
|
size_t size,
|
|
const CUmemAllocationProp *prop,
|
|
unsigned long long flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemRelease (CUmemGenericAllocationHandle handle);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemExportToShareableHandle (void *shareableHandle,
|
|
CUmemGenericAllocationHandle handle,
|
|
CUmemAllocationHandleType handleType,
|
|
unsigned long long flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemImportFromShareableHandle (CUmemGenericAllocationHandle *handle,
|
|
void *osHandle,
|
|
CUmemAllocationHandleType shHandleType);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemSetAccess (CUdeviceptr ptr,
|
|
size_t size,
|
|
const CUmemAccessDesc *desc,
|
|
size_t count);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemGetAccess (unsigned long long *flags,
|
|
const CUmemLocation *location,
|
|
CUdeviceptr ptr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemGetAllocationGranularity (size_t *granularity,
|
|
const CUmemAllocationProp *prop,
|
|
CUmemAllocationGranularity_flags option);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemGetAllocationPropertiesFromHandle (CUmemAllocationProp *prop,
|
|
CUmemGenericAllocationHandle handle);
|
|
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemMap (CUdeviceptr ptr,
|
|
size_t size,
|
|
size_t offset,
|
|
CUmemGenericAllocationHandle handle,
|
|
unsigned long long flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemUnmap (CUdeviceptr ptr,
|
|
size_t size);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemRetainAllocationHandle (CUmemGenericAllocationHandle *handle,
|
|
void *addr);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAllocAsync (CUdeviceptr *dptr,
|
|
size_t bytesize,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemAllocFromPoolAsync (CUdeviceptr *dptr,
|
|
size_t bytesize,
|
|
CUmemoryPool pool,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemFreeAsync (CUdeviceptr dptr,
|
|
CUstream hStream);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemPoolCreate (CUmemoryPool *pool,
|
|
const CUmemPoolProps *poolProps);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemPoolDestroy (CUmemoryPool pool);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemPoolSetAttribute (CUmemoryPool pool,
|
|
CUmemPool_attribute attr,
|
|
void *value);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuMemPoolGetAttribute (CUmemoryPool pool,
|
|
CUmemPool_attribute attr,
|
|
void *value);
|
|
|
|
/* cudaGL.h */
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsGLRegisterImage (CUgraphicsResource * pCudaResource,
|
|
unsigned int image,
|
|
unsigned int target,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsGLRegisterBuffer (CUgraphicsResource * pCudaResource,
|
|
unsigned int buffer,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsResourceSetMapFlags (CUgraphicsResource resource,
|
|
unsigned int flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGLGetDevices (unsigned int * pCudaDeviceCount,
|
|
CUdevice * pCudaDevices,
|
|
unsigned int cudaDeviceCount,
|
|
CUGLDeviceList deviceList);
|
|
|
|
#ifdef G_OS_WIN32
|
|
/* cudaD3D11.h */
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuGraphicsD3D11RegisterResource(CUgraphicsResource * pCudaResource,
|
|
ID3D11Resource * pD3DResource,
|
|
unsigned int Flags);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuD3D11GetDevice(CUdevice * device,
|
|
IDXGIAdapter * pAdapter);
|
|
|
|
GST_CUDA_API
|
|
CUresult CUDAAPI CuD3D11GetDevices(unsigned int * pCudaDeviceCount,
|
|
CUdevice* pCudaDevices,
|
|
unsigned int cudaDeviceCount,
|
|
ID3D11Device * pD3D11Device,
|
|
CUd3d11DeviceList deviceList);
|
|
#endif
|
|
|
|
G_END_DECLS
|