nvml: Load and expose DeviceGetNumGpuCores

2 files changed, 9 insertions, 0 deletions

diff --git a/src/sysinfo/nvml.cpp b/src/sysinfo/nvml.cpp
index 7bea09d..c4ce9d3 100644
--- a/src/sysinfo/nvml.cpp
+++ b/src/sysinfo/nvml.cpp
@@ -27,6 +27,7 @@ namespace dxvk {
         GETPROCADDR(nvmlDeviceGetPerformanceState);
         GETPROCADDR(nvmlDeviceGetUtilizationRates);
         GETPROCADDR(nvmlDeviceGetVbiosVersion);
+        GETPROCADDR(nvmlDeviceGetNumGpuCores);
         GETPROCADDR(nvmlDeviceGetBusType);
         GETPROCADDR(nvmlDeviceGetDynamicPstatesInfo);
 
@@ -112,6 +113,12 @@ namespace dxvk {
             : NVML_ERROR_FUNCTION_NOT_FOUND;
     }
 
+    nvmlReturn_t Nvml::DeviceGetNumGpuCores(nvmlDevice_t device, unsigned int* numCores) const {
+        return m_nvmlDeviceGetNumGpuCores
+            ? m_nvmlDeviceGetNumGpuCores(device, numCores)
+            : NVML_ERROR_FUNCTION_NOT_FOUND;
+    }
+
     nvmlReturn_t Nvml::DeviceGetBusType(nvmlDevice_t device, nvmlBusType_t* type) const {
         return m_nvmlDeviceGetBusType
             ? m_nvmlDeviceGetBusType(device, type)
diff --git a/src/sysinfo/nvml.h b/src/sysinfo/nvml.h
index 6310c82..d8d7a36 100644
--- a/src/sysinfo/nvml.h
+++ b/src/sysinfo/nvml.h
@@ -19,6 +19,7 @@ namespace dxvk {
         [[nodiscard]] virtual nvmlReturn_t DeviceGetPerformanceState(nvmlDevice_t device, nvmlPstates_t* pState) const;
         [[nodiscard]] virtual nvmlReturn_t DeviceGetUtilizationRates(nvmlDevice_t device, nvmlUtilization_t* utilization) const;
         [[nodiscard]] virtual nvmlReturn_t DeviceGetVbiosVersion(nvmlDevice_t device, char* version, unsigned int length) const;
+        [[nodiscard]] virtual nvmlReturn_t DeviceGetNumGpuCores(nvmlDevice_t device, unsigned int* numCores) const;
         [[nodiscard]] virtual nvmlReturn_t DeviceGetBusType(nvmlDevice_t device, nvmlBusType_t* type) const;
         [[nodiscard]] virtual nvmlReturn_t DeviceGetDynamicPstatesInfo(nvmlDevice_t device, nvmlGpuDynamicPstatesInfo_t* pDynamicPstatesInfo) const;
 
@@ -43,6 +44,7 @@ namespace dxvk {
         DECLARE_PFN(nvmlDeviceGetPerformanceState);
         DECLARE_PFN(nvmlDeviceGetUtilizationRates);
         DECLARE_PFN(nvmlDeviceGetVbiosVersion);
+        DECLARE_PFN(nvmlDeviceGetNumGpuCores);
         DECLARE_PFN(nvmlDeviceGetBusType);
         DECLARE_PFN(nvmlDeviceGetDynamicPstatesInfo);