Skip to content

Commit 1fc8cd2

Browse files
committed
refactor: enhance cann variant detection
Signed-off-by: thxCode <[email protected]>
1 parent 7781de6 commit 1fc8cd2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gpustack_runtime/detector/ascend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def _get_device_virtual_info(
342342
"Ascend910_9382": 253,
343343
"Ascend910_9372": 254,
344344
"Ascend910_9362": 255,
345-
"Ascend910_95": 260,
345+
"Ascend910_9579": 260,
346346
}
347347

348348

@@ -415,5 +415,5 @@ def get_ascend_cann_variant(name: str | None) -> str | None:
415415
if version < 260:
416416
return "a3" # 910c
417417
if version < 270:
418-
return "a4" # 910d
418+
return "a5" # 910d
419419
return None

0 commit comments

Comments
 (0)