Skip to content

Commit 451da99

Browse files
Roman ArzumanyanBtbN
authored andcommitted
Add cuCtxGetCurrent function
1 parent 9402b5a commit 451da99

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

include/ffnvcodec/dynlink_cuda.h

Lines changed: 1 addition & 0 deletions
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);

include/ffnvcodec/dynlink_loader.h

Lines changed: 2 additions & 0 deletions
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");

0 commit comments

Comments
 (0)