Add NvlinkInfo.state

Author: mdboom

cuda_core/cuda/core/system/_nvlink.pxi

@@ -20,7 +20,7 @@ cdef class NvlinkInfo:
     @property
     def version(self) -> NvLinkVersion:
         """
-        Retrieves the NvLink version for the device.
+        Retrieves the NvLink version for the device and link.
 
         For all products with NvLink support.
 
@@ -31,4 +31,22 @@ cdef class NvlinkInfo:
         """
         return NvlinkVersion(nvml.device_get_nvlink_version(self._device._handle, self._link))
 
+    @property
+    def state(self) -> bool:
+        """
+        Retrieves the state of the device's NvLink for the device and link specified.
+
+        For Pascal &tm; or newer fully supported devices.
+
+        For all products with NvLink support.
+
+        Returns
+        -------
+
+            `True` if the NvLink is active, `False` otherwise.
+        """
+        return (
+            nvml.device_get_nvlink_state(self._device._handle, self._link) == nvml.EnableState.FEATURE_ENABLED
+        )
+
     max_links = nvml.NVLINK_MAX_LINKS