Skip to content

Commit ab869ff

Browse files
committed
fix: cann 910b detection
Signed-off-by: thxCode <[email protected]>
1 parent d024469 commit ab869ff

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

gpustack_runtime/detector/ascend.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -339,6 +339,7 @@ def _get_device_virtual_info(
339339
"Ascend910_9382": 253,
340340
"Ascend910_9372": 254,
341341
"Ascend910_9362": 255,
342+
"Ascend910_95": 260,
342343
}
343344

344345

@@ -401,6 +402,8 @@ def get_ascend_cann_variant(name: str | None) -> str | None:
401402
if version <= 0:
402403
return None
403404
if version < 200:
405+
if version == 104:
406+
return "910b"
404407
return "910"
405408
if version < 220:
406409
return "310p"
@@ -410,4 +413,6 @@ def get_ascend_cann_variant(name: str | None) -> str | None:
410413
return "310b"
411414
if version < 260:
412415
return "a3" # 910c
416+
if version < 270:
417+
return "a5" # 910d
413418
return None

0 commit comments

Comments
 (0)