feat(nvapi): implement NvAPI_EnumPhysicalGPUs, NvAPI_EnumLogicalGPUs, NvAPI_GetPhysicalGPUsFromDisplay

Author: aquadran

src/nvapi/nvapi.cpp

@@ -265,35 +265,62 @@ NVAPI_INTERFACE
 NvAPI_EnumPhysicalGPUs(NvPhysicalGpuHandle nvGPUHandle[NVAPI_MAX_PHYSICAL_GPUS], NvU32 *pGpuCount) {
   if (!nvGPUHandle || !pGpuCount)
     return NVAPI_INVALID_ARGUMENT;
-  /*
-   * reasonable to report one fake gpu handle
-   */
-  *pGpuCount = 1;
-  nvGPUHandle[0] = (NvPhysicalGpuHandle)0xdeadbeef;
+
+  auto devices = WMT::CopyAllDevices();
+  auto adapterCount = devices.count();
+  if (adapterCount > 1) {
+    *pGpuCount = adapterCount;
+    for (unsigned i = 0; i < adapterCount; i++) {
+      nvGPUHandle[i] = (NvPhysicalGpuHandle)devices.object(i).registryID();
+    }
+  } else {
+    return NVAPI_NVIDIA_DEVICE_NOT_FOUND;
+  }
+
   return NVAPI_OK;
 }
 
 NVAPI_INTERFACE
 NvAPI_EnumLogicalGPUs(NvLogicalGpuHandle nvGPUHandle[NVAPI_MAX_LOGICAL_GPUS], NvU32 *pGpuCount) {
   if (!nvGPUHandle || !pGpuCount)
     return NVAPI_INVALID_ARGUMENT;
-  /*
-   * reasonable to report one fake gpu handle
-   */
-  *pGpuCount = 1;
-  nvGPUHandle[0] = (NvLogicalGpuHandle)0xdeadbeef;
+
+  auto devices = WMT::CopyAllDevices();
+  auto adapterCount = devices.count();
+  if (adapterCount > 1) {
+    *pGpuCount = adapterCount;
+    for (unsigned i = 0; i < adapterCount; i++) {
+      nvGPUHandle[i] = (NvLogicalGpuHandle)devices.object(i).registryID();
+    }
+  } else {
+    return NVAPI_NVIDIA_DEVICE_NOT_FOUND;
+  }
+
   return NVAPI_OK;
 }
 
 NVAPI_INTERFACE
 NvAPI_GetPhysicalGPUsFromDisplay(NvDisplayHandle hNvDisp, NvPhysicalGpuHandle nvGPUHandle[NVAPI_MAX_PHYSICAL_GPUS], NvU32 *pGpuCount) {
-  if (!nvGPUHandle || !pGpuCount)
+  if (!hNvDisp || !nvGPUHandle || !pGpuCount)
     return NVAPI_INVALID_ARGUMENT;
-  /*
-   * reasonable to report one fake gpu handle
-   */
-  *pGpuCount = 1;
-  nvGPUHandle[0] = (NvPhysicalGpuHandle)0xdeadbeef;
+
+  HMONITOR monitor = (HMONITOR)hNvDisp;
+  if (!monitor)
+    return NVAPI_NVIDIA_DEVICE_NOT_FOUND;
+
+  unsigned countDevices = 0;
+  auto devices = WMT::CopyAllDevices();
+  auto adapterCount = devices.count();
+  if (adapterCount > 1) {
+    for (unsigned i = 0; i < adapterCount; i++) {
+      // TODO: Currently assuming all GPU instances are connected to specific monitor
+      nvGPUHandle[countDevices++] = (NvPhysicalGpuHandle)devices.object(i).registryID();
+    }
+  } else {
+    return NVAPI_NVIDIA_DEVICE_NOT_FOUND;
+  }
+  *pGpuCount = countDevices;
+
   return NVAPI_OK;
 }
 
@@ -427,7 +454,6 @@ NvAPI_Disp_HdrColorControl(NvU32 displayId, NV_HDR_COLOR_DATA *pHdrColorData) {
 
 NVAPI_INTERFACE
 NvAPI_EnumNvidiaDisplayHandle(NvU32 thisEnum, NvDisplayHandle *pNvDispHandle) {
-
   if (!pNvDispHandle)
     return NVAPI_INVALID_ARGUMENT;