File tree Expand file tree Collapse file tree
gpustack_runtime/detector Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments