Skip to content

Commit 260c34f

Browse files
committed
Update .pyi files
1 parent 51c5dce commit 260c34f

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

cuda_core/cuda/core/system/_device.pyi

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -787,11 +787,16 @@ class MigInfo:
787787
A list of all MIG devices corresponding to this GPU.
788788
"""
789789

790-
class NvlinkInfo:
790+
class _NvlinkInfoMeta(type):
791+
792+
@property
793+
def max_links(cls):
794+
...
795+
796+
class _NvlinkInfo:
791797
"""
792798
Nvlink information for a device.
793799
"""
794-
max_links = nvml.NVLINK_MAX_LINKS
795800

796801
def __init__(self, device: Device, link: int):
797802
...
@@ -824,6 +829,9 @@ class NvlinkInfo:
824829
`True` if the Nvlink is active.
825830
"""
826831

832+
class NvlinkInfo(_NvlinkInfo, metaclass=_NvlinkInfoMeta):
833+
...
834+
827835
class PciInfo:
828836
"""
829837
PCI information about a GPU device.

0 commit comments

Comments
 (0)