open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/dptech-corp/Uni-Lab-OS.git synced 2026-02-12 18:55:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

添加硬件

Browse Source
This commit is contained in:
zhangshixiang
2025-07-25 12:17:35 +08:00
parent e7d90a8aae
commit ee576c9f6e
13 changed files with 13735 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unilabos/device_mesh/resource_visalization.py
View File

@@ -92,18 +92,18 @@ class ResourceVisualization:
# 遍历设备节点
for node in device.values():
if node['type'] in self.resource_type or (node['type'] == 'device' and node['class'] != ''):
if (node['type'] in self.resource_type and node['class'] != '') or (node['type'] == 'device' and node['class'] != ''):
model_config = {}
if node['type'] in self.resource_type:
resource_class = node['class']
if resource_class not in registry.resource_type_registry.keys():
raise ValueError(f"资源类型 {resource_class} 未在注册表中注册")
raise ValueError(f"{node['id']}资源类型 {resource_class} 未在注册表中注册")
elif "model" in registry.resource_type_registry[resource_class].keys():
model_config = registry.resource_type_registry[resource_class]['model']
elif node['type'] == 'device' and node['class'] != '':
device_class = node['class']
if device_class not in registry.device_type_registry.keys():
raise ValueError(f"设备类型 {device_class} 未在注册表中注册")
raise ValueError(f"{node['id']}设备类型 {device_class} 未在注册表中注册")
elif "model" in registry.device_type_registry[device_class].keys():
model_config = registry.device_type_registry[device_class]['model']
if model_config: