Fix makefile typo (#244)

huggingface · Aug 15, 2024 · 6d9d5ca · 6d9d5ca
1 parent fd1d0f8
commit 6d9d5ca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -122,7 +122,7 @@ install_cli_cuda_onnxruntime:
 # Run tests
 
 test_api_misc:
-	pytest -s -k "api and not (cpu or cuda)
+	pytest -s -k "api and not (cpu or cuda or mps)"
 
 test_api_cpu:
 	pytest -s -k "api and cpu"
@@ -134,7 +134,7 @@ test_api_rocm:
 	pytest -s -k "api and cuda"
 
 test_cli_misc:
-	pytest -s -k "cli and not (cpu or cuda)"
+	pytest -s -k "cli and not (cpu or cuda or mps)"
 
 test_cli_cpu_pytorch:
 	pytest -s -k "cli and cpu and pytorch"

diff --git a/optimum_benchmark/system_utils.py b/optimum_benchmark/system_utils.py
@@ -96,7 +96,8 @@ def get_gpus():
         for i in range(pynvml.nvmlDeviceGetCount()):
             handle = pynvml.nvmlDeviceGetHandleByIndex(i)
             gpu = pynvml.nvmlDeviceGetName(handle)
-            gpu = gpu.decode("utf-8") if isinstance(gpu, bytes) else gpu # Older pynvml may return bytes 
+            # Older pynvml versions may return bytes
+            gpu = gpu.decode("utf-8") if isinstance(gpu, bytes) else gpu
             gpus.append(gpu)
         pynvml.nvmlShutdown()