Skip to content

Commit 2382e15

Browse files
committed
Add cuMemsetD8Async
This function is needed by the vf_thumbnail_cuda filter in ffmpeg.
1 parent 040e692 commit 2382e15

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
@@ -323,6 +323,7 @@ typedef CUresult CUDAAPI tcuCtxPopCurrent_v2(CUcontext *pctx);
323323
typedef CUresult CUDAAPI tcuCtxDestroy_v2(CUcontext ctx);
324324
typedef CUresult CUDAAPI tcuMemAlloc_v2(CUdeviceptr *dptr, size_t bytesize);
325325
typedef CUresult CUDAAPI tcuMemAllocPitch_v2(CUdeviceptr *dptr, size_t *pPitch, size_t WidthInBytes, size_t Height, unsigned int ElementSizeBytes);
326+
typedef CUresult CUDAAPI tcuMemsetD8Async(CUdeviceptr dstDevice, unsigned char uc, size_t N, CUstream hStream);
326327
typedef CUresult CUDAAPI tcuMemFree_v2(CUdeviceptr dptr);
327328
typedef CUresult CUDAAPI tcuMemcpy2D_v2(const CUDA_MEMCPY2D *pcopy);
328329
typedef CUresult CUDAAPI tcuMemcpy2DAsync_v2(const CUDA_MEMCPY2D *pcopy, CUstream hStream);

include/ffnvcodec/dynlink_loader.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@ typedef struct CudaFunctions {
150150
tcuCtxDestroy_v2 *cuCtxDestroy;
151151
tcuMemAlloc_v2 *cuMemAlloc;
152152
tcuMemAllocPitch_v2 *cuMemAllocPitch;
153+
tcuMemsetD8Async *cuMemsetD8Async;
153154
tcuMemFree_v2 *cuMemFree;
154155
tcuMemcpy2D_v2 *cuMemcpy2D;
155156
tcuMemcpy2DAsync_v2 *cuMemcpy2DAsync;
@@ -273,6 +274,7 @@ static inline int cuda_load_functions(CudaFunctions **functions, void *logctx)
273274
LOAD_SYMBOL(cuCtxDestroy, tcuCtxDestroy_v2, "cuCtxDestroy_v2");
274275
LOAD_SYMBOL(cuMemAlloc, tcuMemAlloc_v2, "cuMemAlloc_v2");
275276
LOAD_SYMBOL(cuMemAllocPitch, tcuMemAllocPitch_v2, "cuMemAllocPitch_v2");
277+
LOAD_SYMBOL(cuMemsetD8Async, tcuMemsetD8Async, "cuMemsetD8Async");
276278
LOAD_SYMBOL(cuMemFree, tcuMemFree_v2, "cuMemFree_v2");
277279
LOAD_SYMBOL(cuMemcpy2D, tcuMemcpy2D_v2, "cuMemcpy2D_v2");
278280
LOAD_SYMBOL(cuMemcpy2DAsync, tcuMemcpy2DAsync_v2, "cuMemcpy2DAsync_v2");

0 commit comments

Comments
 (0)