gpustack-runtime 0.1.41.post2__tar.gz → 0.1.42__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/PKG-INFO +2 -2
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/_version.py +2 -2
- gpustack_runtime-0.1.42/gpustack_runtime/_version_appendix.py +1 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/cmds/detector.py +3 -1
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/__types__.py +314 -233
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/__utils__.py +4 -1
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/docker.py +109 -148
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/deviceplugin/__init__.py +21 -3
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/deviceplugin/plugin.py +20 -19
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/kuberentes.py +91 -126
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/podman.py +89 -122
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/__init__.py +2 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/__types__.py +26 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/amd.py +28 -8
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/ascend.py +49 -4
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/cambricon.py +3 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/hygon.py +16 -1
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/iluvatar.py +6 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/metax.py +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/mthreads.py +11 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/nvidia.py +139 -134
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyixml/__init__.py +16 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyrocmsmi/__init__.py +14 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/thead.py +135 -127
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/envs.py +7 -6
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/pyproject.toml +1 -1
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_amd_mi300x.json +2 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_amd_mi308x.json +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_amd_rx7800xt.json +1 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_ascend_310p3.json +2 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_ascend_910b2.json +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_hygon_k100ai.json +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_metax_c500.json +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_gb10.json +1 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_h100.json +8 -0
- gpustack_runtime-0.1.42/tests/gpustack_runtime/detector/samples/detect_output_nvidia_h100_mig.json +309 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_h200.json +8 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_rtx4080super.json +2 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_rtx4090d.json +1 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_nvidia_rtx5090d.json +1 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/detect_output_thead_ppu.json +2 -0
- gpustack_runtime-0.1.42/tests/gpustack_runtime/detector/samples/topology_output_nvidia_h100_mig.json +233 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/uv.lock +4 -4
- gpustack_runtime-0.1.41.post2/gpustack_runtime/_version_appendix.py +0 -1
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.codespelldict +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.codespellrc +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.dockerignore +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.gitattributes +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.gitignore +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.pre-commit-config.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/.python-version +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/LICENSE +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/Makefile +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/README.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/buf.gen.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/buf.lock +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/buf.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/deploy/manifests/docker-compose.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/deploy/manifests/kubernetes.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/docs/index.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/docs/modules/gpustack_runtime.deployer.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/docs/modules/gpustack_runtime.detector.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/docs/modules/gpustack_runtime.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/__main__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/_version.pyi +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/cmds/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/cmds/__types__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/cmds/deployer.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/cmds/images.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/__patches__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/__utils__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/__types__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/amd.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/ascend.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/hygon.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/iluvatar.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/metax.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/cdi/thead.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/deviceplugin/__types__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/api.proto +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/api_pb2.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/api_pb2.pyi +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/api_pb2_grpc.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/deployer/k8s/types/kubelet/deviceplugin/v1beta1/constants.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/__utils__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyacl/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyamdgpu/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyamdsmi/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pycuda/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pydcmi/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyhgml/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyhgml/libhgml.so +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyhgml/libuki.so +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyhsa/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pymxsml/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/detector/pyrocmcore/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/gpustack_runtime/logging.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/hatch.toml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/mkdocs.yml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/pack/Dockerfile +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/pack/Dockerfile.dummy +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/pytest.ini +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/ruff.toml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_compare_versions.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_correct_runner_image.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_load_yaml_or_json.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_load_yaml_or_json_multiple_jsons.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_load_yaml_or_json_multiple_yamls.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_load_yaml_or_json_single_json.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_load_yaml_or_json_single_yaml.yaml +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/fixtures/test_nginx_entrypoint.sh +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/deployer/test_utils.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/fixtures/__init__.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/README.md +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_amd_mi300x.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_amd_mi308x.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_amd_rx7800xt.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_ascend_310p3.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_ascend_910b2.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_hygon_k100ai.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_metax_c500.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_mthreads_s5000.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_nvidia_h100.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_nvidia_h200.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_nvidia_rtx4080super.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_nvidia_rtx4090d.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_nvidia_rtx5090d.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/samples/topology_output_thead_ppu.json +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_amd.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_ascend.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_cambricon.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_hygon.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_iluvatar.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_metax.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_mthreads.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/tests/gpustack_runtime/detector/test_nvidia.py +0 -0
- {gpustack_runtime-0.1.41.post2 → gpustack_runtime-0.1.42}/uv.toml +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: gpustack-runtime
|
|
3
|
-
Version: 0.1.
|
|
3
|
+
Version: 0.1.42
|
|
4
4
|
Summary: GPUStack Runtime is library for detecting GPU resources and launching GPU workloads.
|
|
5
5
|
Project-URL: Homepage, https://github.com/gpustack/runtime
|
|
6
6
|
Project-URL: Bug Tracker, https://github.com/gpustack/gpustack/issues
|
|
@@ -16,7 +16,7 @@ Requires-Python: >=3.10
|
|
|
16
16
|
Requires-Dist: argcomplete>=3.6.3
|
|
17
17
|
Requires-Dist: cachetools>=5.5.2
|
|
18
18
|
Requires-Dist: docker>=7.1.0
|
|
19
|
-
Requires-Dist: gpustack-runner>=0.1.24.
|
|
19
|
+
Requires-Dist: gpustack-runner>=0.1.24.post4
|
|
20
20
|
Requires-Dist: grpc-interceptor>=0.15.4
|
|
21
21
|
Requires-Dist: grpcio>=1.76.0
|
|
22
22
|
Requires-Dist: kubernetes>=33.1.0
|
|
@@ -27,8 +27,8 @@ version_tuple: VERSION_TUPLE
|
|
|
27
27
|
__commit_id__: COMMIT_ID
|
|
28
28
|
commit_id: COMMIT_ID
|
|
29
29
|
|
|
30
|
-
__version__ = version = '0.1.
|
|
31
|
-
__version_tuple__ = version_tuple = (0, 1,
|
|
30
|
+
__version__ = version = '0.1.42'
|
|
31
|
+
__version_tuple__ = version_tuple = (0, 1, 42)
|
|
32
32
|
try:
|
|
33
33
|
from ._version_appendix import git_commit
|
|
34
34
|
__commit_id__ = commit_id = git_commit
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
git_commit = "a9222e0"
|
|
@@ -6,6 +6,7 @@ import time
|
|
|
6
6
|
from typing import TYPE_CHECKING
|
|
7
7
|
|
|
8
8
|
from ..detector import (
|
|
9
|
+
DeviceMemoryStatusEnum,
|
|
9
10
|
Devices,
|
|
10
11
|
Topology,
|
|
11
12
|
detect_devices,
|
|
@@ -142,7 +143,7 @@ def format_devices_table(devs: Devices) -> str:
|
|
|
142
143
|
return "No GPUs detected."
|
|
143
144
|
|
|
144
145
|
# Column headers
|
|
145
|
-
col_headers = ["GPU", "Name", "Memory-Usage", "GPU-Util", "Temp", "CC"]
|
|
146
|
+
col_headers = ["GPU", "Name", "Memory-Usage", "GPU-Util", "Temp", "CC", "Status"]
|
|
146
147
|
# Gather all rows to determine max width for each column
|
|
147
148
|
rows = []
|
|
148
149
|
for dev in devs:
|
|
@@ -153,6 +154,7 @@ def format_devices_table(devs: Devices) -> str:
|
|
|
153
154
|
f"{dev.cores_utilization}%",
|
|
154
155
|
f"{dev.temperature}C" if dev.temperature is not None else "N/A",
|
|
155
156
|
dev.compute_capability if dev.compute_capability else "N/A",
|
|
157
|
+
"OK" if dev.memory_status == DeviceMemoryStatusEnum.HEALTHY else "ERR",
|
|
156
158
|
]
|
|
157
159
|
rows.append(row)
|
|
158
160
|
|