Address comments in the PR

Author: mdboom

cuda_core/cuda/core/system/_device.pyx

@@ -775,7 +775,7 @@ cdef class Device:
         Querying per-instance information using MIG device handles is not
         supported if the device is in vGPU Host virtualization mode.
         """
-        return [ProcessInfo(proc) for proc in nvml.device_get_compute_running_processes_v3(self._handle)]
+        return [ProcessInfo(self, proc) for proc in nvml.device_get_compute_running_processes_v3(self._handle)]
 
     ##########################################################################
     # REPAIR STATUS

cuda_core/cuda/core/system/_process.pxi

@@ -5,9 +5,10 @@
 
 class ProcessInfo:
     """
-    Information about a process using the GPU.
+    Information about running compute processes on the GPU.
     """
-    def __init__(self, process_info: nvml.ProcessInfo):
+    def __init__(self, device: "Device", process_info: nvml.ProcessInfo):
+        self._device = device
         self._process_info = process_info
 
     @property
@@ -31,6 +32,10 @@ class ProcessInfo:
 
         Only valid for processes running on MIG devices.
         """
+        if not self._device.mig.is_mig_device:
+            raise nvml.UnsupportedError(
+                "GPU instance ID is only valid for processes running on MIG devices."
+            )
         return self._process_info.gpu_instance_id
 
     @property
@@ -40,4 +45,8 @@ class ProcessInfo:
 
         Only valid for processes running on MIG devices.
         """
+        if not self._device.mig.is_mig_device:
+            raise nvml.UnsupportedError(
+                "Compute instance ID is only valid for processes running on MIG devices."
+            )
         return self._process_info.compute_instance_id

cuda_core/tests/system/test_system_device.py

@@ -740,8 +740,14 @@ def test_compute_running_processes():
             assert isinstance(proc, _device.ProcessInfo)
             assert isinstance(proc.pid, int)
             assert isinstance(proc.used_gpu_memory, int)
-            assert isinstance(proc.gpu_instance_id, int)
-            assert isinstance(proc.compute_instance_id, int)
+            if device.mig.is_mig_device:
+                assert isinstance(proc.gpu_instance_id, int)
+                assert isinstance(proc.compute_instance_id, int)
+            else:
+                with pytest.raises(nvml.UnsupportedError):
+                    proc.gpu_instance_id  # noqa: B018
+                with pytest.raises(nvml.UnsupportedError):
+                    proc.compute_instance_id  # noqa: B018
 
 
 @pytest.mark.skipif(helpers.IS_WSL or helpers.IS_WINDOWS, reason="MIG not supported on WSL or Windows")