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

fix(gui/device): fix MIG device support

Browse source
This commit is contained in:
Xuehai Pan 2023-04-07 14:37:34 +00:00
parent 790ffdf404
commit 1616a486c9
1 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
nvitop/gui/library/device.py
View file

@ -80,6 +80,20 @@ class Device(DeviceBase):
self.as_snapshot()
return self._snapshot
def mig_devices(self):
mig_devices = []
if self.is_mig_mode_enabled():
for mig_index in range(self.max_mig_device_count()):
try:
mig_device = MigDevice(index=(self.index, mig_index))
except libnvml.NVMLError:
break
else:
mig_devices.append(mig_device)
return mig_devices
fan_speed = ttl_cache(ttl=5.0)(DeviceBase.fan_speed)
temperature = ttl_cache(ttl=5.0)(DeviceBase.temperature)
power_usage = ttl_cache(ttl=5.0)(DeviceBase.power_usage)