Add cuCtxGetCurrent function · FFmpeg/nv-codec-headers@625b319

2 files changed

lines changed

Original file line numberDiff line numberDiff line change

@@ -428,6 +428,7 @@ typedef CUresult CUDAAPI tcuDeviceGetByPCIBusId(CUdevice* dev, const char* pciBu

428428

typedef CUresult CUDAAPI tcuDeviceGetPCIBusId(char* pciBusId, int len, CUdevice dev);

429429

typedef CUresult CUDAAPI tcuDeviceComputeCapability(int *major, int *minor, CUdevice dev);

430430

typedef CUresult CUDAAPI tcuCtxCreate_v2(CUcontext *pctx, unsigned int flags, CUdevice dev);

431+

typedef CUresult CUDAAPI tcuCtxGetCurrent(CUcontext *pctx);

431432

typedef CUresult CUDAAPI tcuCtxSetLimit(CUlimit limit, size_t value);

432433

typedef CUresult CUDAAPI tcuCtxPushCurrent_v2(CUcontext pctx);

433434

typedef CUresult CUDAAPI tcuCtxPopCurrent_v2(CUcontext *pctx);

Original file line numberDiff line numberDiff line change

@@ -149,6 +149,7 @@ typedef struct CudaFunctions {

149149

tcuDeviceGetPCIBusId *cuDeviceGetPCIBusId;

150150

tcuDeviceComputeCapability *cuDeviceComputeCapability;

151151

tcuCtxCreate_v2 *cuCtxCreate;

152+

tcuCtxGetCurrent *cuCtxGetCurrent;

152153

tcuCtxSetLimit *cuCtxSetLimit;

153154

tcuCtxPushCurrent_v2 *cuCtxPushCurrent;

154155

tcuCtxPopCurrent_v2 *cuCtxPopCurrent;

@@ -315,6 +316,7 @@ static inline int cuda_load_functions(CudaFunctions **functions, void *logctx)

315316

LOAD_SYMBOL(cuDeviceGetName, tcuDeviceGetName, "cuDeviceGetName");

316317

LOAD_SYMBOL(cuDeviceComputeCapability, tcuDeviceComputeCapability, "cuDeviceComputeCapability");

317318

LOAD_SYMBOL(cuCtxCreate, tcuCtxCreate_v2, "cuCtxCreate_v2");

319+

LOAD_SYMBOL(cuCtxGetCurrent, tcuCtxGetCurrent, "cuCtxGetCurrent");

318320

LOAD_SYMBOL(cuCtxSetLimit, tcuCtxSetLimit, "cuCtxSetLimit");

319321

LOAD_SYMBOL(cuCtxPushCurrent, tcuCtxPushCurrent_v2, "cuCtxPushCurrent_v2");

320322

LOAD_SYMBOL(cuCtxPopCurrent, tcuCtxPopCurrent_v2, "cuCtxPopCurrent_v2");