Skip to content

Commit f1f32e7

Browse files
committed
fix: wrong collect memory in ascend
Signed-off-by: thxCode <[email protected]>
1 parent 995aae8 commit f1f32e7

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

gpustack_runtime/detector/ascend.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,9 @@ def _get_device_memory_info(dev_card_id, dev_device_id) -> tuple[int, int]:
227227
dev_device_id,
228228
)
229229
dev_mem = dev_memory_info.memory_size
230-
dev_mem_used = dev_memory_info.utiliza
230+
dev_mem_used = (
231+
dev_memory_info.memory_size - dev_memory_info.memory_available
232+
)
231233
except pydcmi.DCMIError as e:
232234
if e.value in [
233235
pydcmi.DCMI_ERROR_FUNCTION_NOT_FOUND,
@@ -239,7 +241,9 @@ def _get_device_memory_info(dev_card_id, dev_device_id) -> tuple[int, int]:
239241
dev_device_id,
240242
)
241243
dev_mem = dev_memory_info.memory_size
242-
dev_mem_used = dev_memory_info.utiliza
244+
dev_mem_used = (
245+
dev_memory_info.memory_size - dev_memory_info.memory_available
246+
)
243247
else:
244248
raise
245249

0 commit comments

Comments
 (0)