zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-02 01:08:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix missing domain_uuid_key in thunderbolt ring setup (#2682)

Browse Source
This commit is contained in:
Christopher Webb
2025-10-30 15:17:20 -05:00
committed by GitHub
parent 74c1ed25bb
commit 793a31eeb6
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/mlx/distributed_run.py
View File

@@ -567,13 +567,16 @@ def prepare_tb_ring(args, hosts):
name = "" name = ""
ports = [] ports = []
for t in c["SPThunderboltDataType"]: for t in c["SPThunderboltDataType"]:
uuid = t.get("domain_uuid_key")
if uuid is None:
continue
name = t["device_name_key"] name = t["device_name_key"]
uuid = t["domain_uuid_key"]
tag = t["receptacle_1_tag"]["receptacle_id_key"] tag = t["receptacle_1_tag"]["receptacle_id_key"]
if items := t.get("_items", []): items = t.get("_items", [])
connected_to = items[0]["domain_uuid_key"] connected_items = [item for item in items if "domain_uuid_key" in item]
else: connected_to = (
connected_to = None connected_items[0]["domain_uuid_key"] if connected_items else None
)
iface = iface_map[f"Thunderbolt {tag}"] iface = iface_map[f"Thunderbolt {tag}"]
ports.append(ThunderboltPort(iface, uuid, connected_to)) ports.append(ThunderboltPort(iface, uuid, connected_to))
tb_hosts.append(ThunderboltHost(name, sorted(ports, key=lambda x: x.iface))) tb_hosts.append(ThunderboltHost(name, sorted(ports, key=lambda x: x.iface)))