github-starred/nvitop
2
0
Fork 0
mirror of https://github.com/XuehaiPan/nvitop.git synced 2026-05-21 06:45:24 -06:00
Code Issues 29 Projects Packages Wiki Activity Actions

feat(api/device): add shortcut when there is no NVLink available

Browse source
This commit is contained in:
Xuehai Pan 2023-08-05 02:23:29 +08:00
parent 4073d7da75
commit b2dc47eec8
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nvitop/api/device.py
View file

@ -1537,6 +1537,8 @@ class Device: # pylint: disable=too-many-instance-attributes,too-many-public-me
could be :const:`nvitop.NA` when not applicable.
"""
nvlink_link_count = self.nvlink_link_count()
if nvlink_link_count == 0:
return []
def query_nvlink_throughput_counters() -> tuple[tuple[int | NaType, int]]:
return tuple( # type: ignore[return-value]