Skip to content

Commit dee9a2f

Browse files
authored
Merge pull request #2 from Readon/copilot/fix-ruff-formatting-issues
Fix ruff format violations in AMD APU unified memory VRAM detection
2 parents 8933257 + a5b16bd commit dee9a2f

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

  • gpustack_runtime/detector

gpustack_runtime/detector/amd.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -185,8 +185,9 @@ def detect(self) -> Devices | None:
185185
with contextlib.suppress(pyrocmsmi.ROCMSMIError):
186186
dev_mem_vis_vram = byte_to_mebibyte(
187187
pyrocmsmi.rsmi_dev_memory_total_get(
188-
dev_idx, pyrocmsmi.RSMI_MEM_TYPE_VIS_VRAM,
189-
),
188+
dev_idx,
189+
pyrocmsmi.RSMI_MEM_TYPE_VIS_VRAM,
190+
)
190191
)
191192
dev_mem = max(dev_mem, dev_mem_vis_vram)
192193
dev_ecc_count = pyamdsmi.amdsmi_get_gpu_ecc_count(
@@ -205,8 +206,9 @@ def detect(self) -> Devices | None:
205206
with contextlib.suppress(pyrocmsmi.ROCMSMIError):
206207
dev_mem_vis_vram = byte_to_mebibyte(
207208
pyrocmsmi.rsmi_dev_memory_total_get(
208-
dev_idx, pyrocmsmi.RSMI_MEM_TYPE_VIS_VRAM,
209-
),
209+
dev_idx,
210+
pyrocmsmi.RSMI_MEM_TYPE_VIS_VRAM,
211+
)
210212
)
211213
dev_mem = max(dev_mem, dev_mem_vis_vram)
212214
dev_mem_used = byte_to_mebibyte( # byte to MiB

0 commit comments

Comments
 (0)