gpustack-runtime 0.1.41.post2__py3-none-any.whl → 0.1.42__py3-none-any.whl

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.
Files changed (29) hide show
  1. gpustack_runtime/_version.py +2 -2
  2. gpustack_runtime/_version_appendix.py +1 -1
  3. gpustack_runtime/cmds/detector.py +3 -1
  4. gpustack_runtime/deployer/__types__.py +314 -233
  5. gpustack_runtime/deployer/cdi/__utils__.py +4 -1
  6. gpustack_runtime/deployer/docker.py +109 -148
  7. gpustack_runtime/deployer/k8s/deviceplugin/__init__.py +21 -3
  8. gpustack_runtime/deployer/k8s/deviceplugin/plugin.py +20 -19
  9. gpustack_runtime/deployer/kuberentes.py +91 -126
  10. gpustack_runtime/deployer/podman.py +89 -122
  11. gpustack_runtime/detector/__init__.py +2 -0
  12. gpustack_runtime/detector/__types__.py +26 -0
  13. gpustack_runtime/detector/amd.py +28 -8
  14. gpustack_runtime/detector/ascend.py +49 -4
  15. gpustack_runtime/detector/cambricon.py +3 -0
  16. gpustack_runtime/detector/hygon.py +16 -1
  17. gpustack_runtime/detector/iluvatar.py +6 -0
  18. gpustack_runtime/detector/metax.py +8 -0
  19. gpustack_runtime/detector/mthreads.py +11 -0
  20. gpustack_runtime/detector/nvidia.py +139 -134
  21. gpustack_runtime/detector/pyixml/__init__.py +16 -0
  22. gpustack_runtime/detector/pyrocmsmi/__init__.py +14 -0
  23. gpustack_runtime/detector/thead.py +135 -127
  24. gpustack_runtime/envs.py +7 -6
  25. {gpustack_runtime-0.1.41.post2.dist-info → gpustack_runtime-0.1.42.dist-info}/METADATA +2 -2
  26. {gpustack_runtime-0.1.41.post2.dist-info → gpustack_runtime-0.1.42.dist-info}/RECORD +29 -29
  27. {gpustack_runtime-0.1.41.post2.dist-info → gpustack_runtime-0.1.42.dist-info}/WHEEL +0 -0
  28. {gpustack_runtime-0.1.41.post2.dist-info → gpustack_runtime-0.1.42.dist-info}/entry_points.txt +0 -0
  29. {gpustack_runtime-0.1.41.post2.dist-info → gpustack_runtime-0.1.42.dist-info}/licenses/LICENSE +0 -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.41.post2'
31
- __version_tuple__ = version_tuple = (0, 1, 41, 'post2')
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
@@ -1 +1 @@
1
- git_commit = "ed331dc"
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