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

chore(select): show only one MIG when calling nvisel without arguments

Browse source 
Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-08-05 14:24:02 +08:00
parent e688b44fa7
commit c2193768b3
1 changed files with 6 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

15
nvitop/select.py
View file

@ -230,15 +230,12 @@ def select_devices(
)
if any(device.is_mig_device for device in available_devices): # found MIG devices!
if min_count >= 2:
non_mig_devices = [device for device in available_devices if not device.is_mig_device]
mig_devices = [device for device in available_devices if device.is_mig_device]
if len(non_mig_devices) >= min_count or not available_devices[0].is_mig_device:
available_devices = non_mig_devices
else:
available_devices = mig_devices[:1] # at most one MIG device is visible
# Check again
if any(device.is_mig_device for device in available_devices): # found MIG devices!
non_mig_devices = [device for device in available_devices if not device.is_mig_device]
mig_devices = [device for device in available_devices if device.is_mig_device]
if len(non_mig_devices) >= min_count > 0 or not available_devices[0].is_mig_device:
available_devices = non_mig_devices
else:
available_devices = mig_devices[:1] # at most one MIG device is visible
if format == 'index' and not force_index:
format = 'uuid'