Skip to content

Commit 035f427

Browse files
committed
fix(ascend): failed to detect a3
Signed-off-by: thxCode <thxcode0824@gmail.com>
1 parent 03ad13b commit 035f427

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

gpustack_runtime/detector/ascend.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -444,7 +444,7 @@ def _get_device_roce_network_info(
444444
pydcmi.DCMI_PORT_TYPE_ROCE_PORT,
445445
)
446446
except pydcmi.DCMIError:
447-
debug_log_exception(logger, "Failed to get device roce network info")
447+
debug_log_exception(logger, "Failed to get device RoCE network info")
448448

449449
return ip, mask, gateway
450450

@@ -505,12 +505,15 @@ def _get_device_virtual_info(
505505
"Ascend310B3": 242,
506506
"Ascend310B4": 243,
507507
"Ascend910_9391": 250,
508+
"Ascend910": 250,
508509
"Ascend910_9392": 251,
509510
"Ascend910_9381": 252,
510511
"Ascend910_9382": 253,
511512
"Ascend910_9372": 254,
512513
"Ascend910_9362": 255,
513514
"Ascend910_9579": 260,
515+
"Ascend910_95": 260,
516+
"Ascend950": 260,
514517
}
515518

516519

@@ -526,6 +529,8 @@ def _guess_soc_name_from_dev_name(dev_name: str) -> str | None:
526529
The guessed SoC name, or None if not found.
527530
528531
"""
532+
if dev_name.startswith("Ascend"):
533+
dev_name = dev_name[6:].strip()
529534
soc_name = f"Ascend{dev_name}"
530535
if soc_name in _soc_name_version_mapping:
531536
return soc_name

0 commit comments

Comments
 (0)