Signed-off-by: softworkz
---
include/ffnvcodec/dynlink_cuda.h | 26 ++
include/ffnvcodec/dynlink_loader.h | 2 ++
2 files changed, 28 insertions(+)
diff --git a/include/ffnvcodec/dynlink_cuda.h b/include/ffnvcodec/dynlink_cuda.h
index 373215d..069acd1 100644
--- a/include/ffnvcodec/dynlink_cuda.h
+++ b/include/ffnvcodec/dynlink_cuda.h
@@ -59,6 +59,31 @@ typedef enum cudaError_enum {
CUDA_ERROR_NOT_READY = 600
} CUresult;
+/**
+ * Device properties (subset)
+ */
+typedef enum CUdevice_attribute_enum {
+CU_DEVICE_ATTRIBUTE_CLOCK_RATE = 13,
+CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT = 16,
+CU_DEVICE_ATTRIBUTE_INTEGRATED = 18,
+CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY = 19,
+CU_DEVICE_ATTRIBUTE_COMPUTE_MODE = 20,
+CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS = 31,
+CU_DEVICE_ATTRIBUTE_PCI_BUS_ID = 33,
+CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID = 34,
+CU_DEVICE_ATTRIBUTE_TCC_DRIVER = 35,
+CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE = 36,
+CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH = 37,
+CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT = 40,
+CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING = 41,
+CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID = 50,
+CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR = 75,
+CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR = 76,
+CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY = 83,
+CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD = 84,
+CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD_GROUP_ID = 85,
+} CUdevice_attribute;
+
typedef enum CUarray_format_enum {
CU_AD_FORMAT_UNSIGNED_INT8 = 0x01,
CU_AD_FORMAT_UNSIGNED_INT16 = 0x02,
@@ -184,6 +209,7 @@ typedef void CUDAAPI CUstreamCallback(CUstream hStream,
CUresult status, void *u
typedef CUresult CUDAAPI tcuInit(unsigned int Flags);
typedef CUresult CUDAAPI tcuDeviceGetCount(int *count);
typedef CUresult CUDAAPI tcuDeviceGet(CUdevice *device, int ordinal);
+typedef CUresult CUDAAPI tcuDeviceGetAttribute(int *pi, CUdevice_attribute
attrib, CUdevice dev);
typedef CUresult CUDAAPI tcuDeviceGetName(char *name, int len, CUdevice dev);
typedef CUresult CUDAAPI tcuDeviceGetUuid(CUuuid *uuid, CUdevice dev);
typedef CUresult CUDAAPI tcuDeviceComputeCapability(int *major, int *minor,
CUdevice dev);
diff --git a/include/ffnvcodec/dynlink_loader.h
b/include/ffnvcodec/dynlink_loader.h
index bce9630..c9472fe 100644
--- a/include/ffnvcodec/dynlink_loader.h
+++ b/include/ffnvcodec/dynlink_loader.h
@@ -139,6 +139,7 @@ typedef struct CudaFunctions {
tcuInit *cuInit;
tcuDeviceGetCount *cuDeviceGetCount;
tcuDeviceGet *cuDeviceGet;
+tcuDeviceGetAttribute *cuDeviceGetAttribute;
tcuDeviceGetName *cuDeviceGetName;
tcuDeviceGetUuid *cuDeviceGetUuid;
tcuDeviceComputeCapability *cuDeviceComputeCapability;
@@ -248,6 +249,7 @@ static inline int cuda_load_functions(CudaFunctions
**functions, void *logctx)
LOAD_SYMBOL(cuInit, tcuInit, "cuInit");
LOAD_SYMBOL(cuDeviceGetCount, tcuDeviceGetCount, "cuDeviceGetCount");
LOAD_SYMBOL(cuDeviceGet, tcuDeviceGet, "cuDeviceGet");
+LOAD_SYMBOL(cuDeviceGetAttribute, tcuDeviceGetAttribute,
"cuDeviceGetAttribute");
LOAD_SYMBOL(cuDeviceGetName, tcuDeviceGetName, "cuDeviceGetName");
LOAD_SYMBOL(cuDeviceGetUuid, tcuDeviceGetUuid, "cuDeviceGetUuid");
LOAD_SYMBOL(cuDeviceComputeCapability, tcuDeviceComputeCapability,
"cuDeviceComputeCapability");
--
2.17.1.windows.2
___
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
http://ffmpeg.org/mailman/listinfo/ffmpeg-devel