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 @@ -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
You can’t perform that action at this time.
0 commit comments