common
zeus.device.gpu.common
Error wrappers and classes common to all GPU vendors.
GPU
Bases: ABC
Abstract base class for managing one GPU.
For each method, child classes should call into vendor-specific GPU management libraries (e.g., NVML for NVIDIA GPUs).
Source code in zeus/device/gpu/common.py
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
|
supports_nonblocking_setters
abstractmethod
property
supports_nonblocking_setters
Return True if the GPU object supports non-blocking configuration setters.
__init__
__init__(gpu_index)
Source code in zeus/device/gpu/common.py
23 24 25 |
|
getName
abstractmethod
getName()
Return the name of the GPU model.
Source code in zeus/device/gpu/common.py
33 34 35 36 |
|
getPowerManagementLimitConstraints
abstractmethod
getPowerManagementLimitConstraints()
Return the minimum and maximum power management limits. Units: mW.
Source code in zeus/device/gpu/common.py
38 39 40 41 |
|
setPowerManagementLimit
abstractmethod
setPowerManagementLimit(power_limit_mw, _block=True)
Set the GPU's power management limit. Unit: mW.
Source code in zeus/device/gpu/common.py
43 44 45 46 |
|
resetPowerManagementLimit
abstractmethod
resetPowerManagementLimit(_block=True)
Reset the GPU's power management limit to the default value.
Source code in zeus/device/gpu/common.py
48 49 50 51 |
|
setPersistenceMode
abstractmethod
setPersistenceMode(enabled, _block=True)
Set persistence mode.
Source code in zeus/device/gpu/common.py
53 54 55 56 |
|
getSupportedMemoryClocks
abstractmethod
getSupportedMemoryClocks()
Return a list of supported memory clock frequencies. Units: MHz.
Source code in zeus/device/gpu/common.py
58 59 60 61 |
|
setMemoryLockedClocks
abstractmethod
setMemoryLockedClocks(
min_clock_mhz, max_clock_mhz, _block=True
)
Lock the memory clock to a specified range. Units: MHz.
Source code in zeus/device/gpu/common.py
63 64 65 66 67 68 |
|
resetMemoryLockedClocks
abstractmethod
resetMemoryLockedClocks(_block=True)
Reset the locked memory clocks to the default.
Source code in zeus/device/gpu/common.py
70 71 72 73 |
|
getSupportedGraphicsClocks
abstractmethod
getSupportedGraphicsClocks(memory_clock_mhz=None)
Return a list of supported graphics clock frequencies. Units: MHz.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
memory_clock_mhz |
int | None
|
Memory clock frequency to use. Some GPUs have different supported graphics clocks depending on the memory clock. |
None
|
Source code in zeus/device/gpu/common.py
75 76 77 78 79 80 81 82 83 84 85 |
|
setGpuLockedClocks
abstractmethod
setGpuLockedClocks(
min_clock_mhz, max_clock_mhz, _block=True
)
Lock the GPU clock to a specified range. Units: MHz.
Source code in zeus/device/gpu/common.py
87 88 89 90 91 92 |
|
resetGpuLockedClocks
abstractmethod
resetGpuLockedClocks(_block=True)
Reset the locked GPU clocks to the default.
Source code in zeus/device/gpu/common.py
94 95 96 97 |
|
getInstantPowerUsage
abstractmethod
getInstantPowerUsage()
Return the current power draw of the GPU. Units: mW.
Source code in zeus/device/gpu/common.py
99 100 101 102 |
|
getAverageMemoryPowerUsage
abstractmethod
getAverageMemoryPowerUsage()
Return the average power usage of the GPU's memory. Units: mW.
Source code in zeus/device/gpu/common.py
104 105 106 107 |
|
supportsGetTotalEnergyConsumption
abstractmethod
supportsGetTotalEnergyConsumption()
Check if the GPU supports retrieving total energy consumption.
Source code in zeus/device/gpu/common.py
109 110 111 112 |
|
getTotalEnergyConsumption
abstractmethod
getTotalEnergyConsumption()
Return the total energy consumption of the GPU since driver load. Units: mJ.
Source code in zeus/device/gpu/common.py
114 115 116 117 |
|
GPUs
Bases: ABC
An abstract base class for a collection of GPU
objects.
This is basically a list of GPU
objects and forwards
most API calls to the individual GPU
objects. Still, a separate wrapper class is
is needed to for group-level operations like:
ensure_homogeneous
that ensures that all GPUs have the same name- handling vendor-specific environment variables (e.g.,
CUDA_VISIBLE_DEVICES
for NVIDIA)
Source code in zeus/device/gpu/common.py
120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 |
|
gpus
abstractmethod
property
gpus
Return a list of GPU objects being tracked.
__init__
abstractmethod
__init__(ensure_homogeneous=False)
Source code in zeus/device/gpu/common.py
131 132 133 134 |
|
__del__
abstractmethod
__del__()
Shut down the GPU monitoring library to release resources and clean up.
Source code in zeus/device/gpu/common.py
136 137 138 139 |
|
__len__
__len__()
Return the number of GPUs being tracked.
Source code in zeus/device/gpu/common.py
147 148 149 |
|
_ensure_homogeneous
_ensure_homogeneous()
Ensures that all tracked GPUs are homogeneous in terms of name.
Source code in zeus/device/gpu/common.py
151 152 153 154 155 156 |
|
_warn_sys_admin
_warn_sys_admin()
Warn the user if the current process doesn't have SYS_ADMIN
privileges.
Source code in zeus/device/gpu/common.py
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
|
getName
getName(gpu_index)
Return the name of the specified GPU.
Source code in zeus/device/gpu/common.py
176 177 178 |
|
getPowerManagementLimitConstraints
getPowerManagementLimitConstraints(gpu_index)
Return the minimum and maximum power management limits. Units: mW.
Source code in zeus/device/gpu/common.py
180 181 182 |
|
setPowerManagementLimit
setPowerManagementLimit(
gpu_index, power_limit_mw, _block=True
)
Set the GPU's power management limit. Unit: mW.
Source code in zeus/device/gpu/common.py
184 185 186 187 188 189 |
|
resetPowerManagementLimit
resetPowerManagementLimit(gpu_index, _block=True)
Reset the GPU's power management limit to the default value.
Source code in zeus/device/gpu/common.py
191 192 193 194 |
|
setPersistenceMode
setPersistenceMode(gpu_index, enabled, _block=True)
Set persistence mode for the specified GPU.
Source code in zeus/device/gpu/common.py
196 197 198 199 200 201 |
|
getSupportedMemoryClocks
getSupportedMemoryClocks(gpu_index)
Return a list of supported memory clock frequencies. Units: MHz.
Source code in zeus/device/gpu/common.py
203 204 205 |
|
setMemoryLockedClocks
setMemoryLockedClocks(
gpu_index, min_clock_mhz, max_clock_mhz, _block=True
)
Lock the memory clock to a specified range. Units: MHz.
Source code in zeus/device/gpu/common.py
207 208 209 210 211 212 213 214 215 216 |
|
resetMemoryLockedClocks
resetMemoryLockedClocks(gpu_index, _block=True)
Reset the locked memory clocks to the default.
Source code in zeus/device/gpu/common.py
218 219 220 221 |
|
getSupportedGraphicsClocks
getSupportedGraphicsClocks(
gpu_index, memory_clock_mhz=None
)
Return a list of supported graphics clock frequencies. Units: MHz.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
gpu_index |
int
|
Index of the GPU to query. |
required |
memory_clock_mhz |
int | None
|
Memory clock frequency to use. Some GPUs have different supported graphics clocks depending on the memory clock. |
None
|
Source code in zeus/device/gpu/common.py
223 224 225 226 227 228 229 230 231 232 233 |
|
setGpuLockedClocks
setGpuLockedClocks(
gpu_index, min_clock_mhz, max_clock_mhz, _block=True
)
Lock the GPU clock to a specified range. Units: MHz.
Source code in zeus/device/gpu/common.py
235 236 237 238 239 240 241 242 243 244 |
|
resetGpuLockedClocks
resetGpuLockedClocks(gpu_index, _block=True)
Reset the locked GPU clocks to the default.
Source code in zeus/device/gpu/common.py
246 247 248 249 |
|
getInstantPowerUsage
getInstantPowerUsage(gpu_index)
Return the current power draw of the GPU. Units: mW.
Source code in zeus/device/gpu/common.py
251 252 253 |
|
getAverageMemoryPowerUsage
getAverageMemoryPowerUsage(gpu_index)
Return the average power usage of the GPU's memory. Units: mW.
Source code in zeus/device/gpu/common.py
255 256 257 |
|
supportsGetTotalEnergyConsumption
supportsGetTotalEnergyConsumption(gpu_index)
Check if the GPU supports retrieving total energy consumption.
Source code in zeus/device/gpu/common.py
259 260 261 |
|
getTotalEnergyConsumption
getTotalEnergyConsumption(gpu_index)
Return the total energy consumption of the GPU since driver load. Units: mJ.
Source code in zeus/device/gpu/common.py
263 264 265 |
|
EmptyGPUs
Bases: GPUs
A concrete class implementing the GPUs abstract base class, but representing an empty collection of GPUs.
This class is used to represent a scenario where no GPUs are available or detected. Any method call attempting to interact with a GPU will raise a ValueError.
Source code in zeus/device/gpu/common.py
268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 |
|
gpus
property
gpus
Return an empty list as no GPUs are being tracked.
__init__
__init__(ensure_homogeneous=False)
Since this class represents an empty collection of GPUs, no actual initialization of GPU objects is performed.
Source code in zeus/device/gpu/common.py
275 276 277 278 279 280 |
|
__del__
__del__()
Clean up any resources if necessary.
As this class represents an empty collection of GPUs, no specific cleanup is required.
Source code in zeus/device/gpu/common.py
282 283 284 285 286 287 |
|
__len__
__len__()
Return 0, indicating no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
294 295 296 |
|
_ensure_homogeneous
_ensure_homogeneous()
Raise a ValueError as no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
298 299 300 |
|
_warn_sys_admin
_warn_sys_admin()
Raise a ValueError as no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
302 303 304 |
|
getName
getName(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
306 307 308 |
|
getPowerManagementLimitConstraints
getPowerManagementLimitConstraints(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
310 311 312 |
|
setPowerManagementLimit
setPowerManagementLimit(
gpu_index, power_limit_mw, _block=True
)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
314 315 316 317 318 |
|
resetPowerManagementLimit
resetPowerManagementLimit(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
320 321 322 |
|
setPersistenceMode
setPersistenceMode(gpu_index, enabled, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
324 325 326 327 328 |
|
getSupportedMemoryClocks
getSupportedMemoryClocks(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
330 331 332 |
|
setMemoryLockedClocks
setMemoryLockedClocks(
gpu_index, min_clock_mhz, max_clock_mhz, _block=True
)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
334 335 336 337 338 339 340 341 342 |
|
resetMemoryLockedClocks
resetMemoryLockedClocks(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
344 345 346 |
|
getSupportedGraphicsClocks
getSupportedGraphicsClocks(
gpu_index, memory_clock_mhz=None
)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
348 349 350 351 352 |
|
setGpuLockedClocks
setGpuLockedClocks(
gpu_index, min_clock_mhz, max_clock_mhz, _block=True
)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
354 355 356 357 358 359 360 361 362 |
|
resetGpuLockedClocks
resetGpuLockedClocks(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
364 365 366 |
|
getInstantPowerUsage
getInstantPowerUsage(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
368 369 370 |
|
supportsGetTotalEnergyConsumption
supportsGetTotalEnergyConsumption(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
372 373 374 |
|
getTotalEnergyConsumption
getTotalEnergyConsumption(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
376 377 378 |
|
ZeusGPUInitError
Bases: ZeusBaseGPUError
Import error or GPU library initialization failures.
Source code in zeus/device/gpu/common.py
381 382 383 384 385 386 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
384 385 386 |
|
ZeusGPUInvalidArgError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Invalid Argument.
Source code in zeus/device/gpu/common.py
389 390 391 392 393 394 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
392 393 394 |
|
ZeusGPUNotSupportedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Not Supported Operation on GPU.
Source code in zeus/device/gpu/common.py
397 398 399 400 401 402 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
400 401 402 |
|
ZeusGPUNoPermissionError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps No Permission to perform GPU operation.
Source code in zeus/device/gpu/common.py
405 406 407 408 409 410 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
408 409 410 |
|
ZeusGPUAlreadyInitializedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Already Initialized GPU.
Source code in zeus/device/gpu/common.py
413 414 415 416 417 418 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
416 417 418 |
|
ZeusGPUNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Not Found GPU.
Source code in zeus/device/gpu/common.py
421 422 423 424 425 426 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
424 425 426 |
|
ZeusGPUInsufficientSizeError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Size.
Source code in zeus/device/gpu/common.py
429 430 431 432 433 434 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
432 433 434 |
|
ZeusGPUInsufficientPowerError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Power.
Source code in zeus/device/gpu/common.py
437 438 439 440 441 442 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
440 441 442 |
|
ZeusGPUDriverNotLoadedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Driver Error.
Source code in zeus/device/gpu/common.py
445 446 447 448 449 450 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
448 449 450 |
|
ZeusGPUTimeoutError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Timeout Error.
Source code in zeus/device/gpu/common.py
453 454 455 456 457 458 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
456 457 458 |
|
ZeusGPUIRQError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps IRQ Error.
Source code in zeus/device/gpu/common.py
461 462 463 464 465 466 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
464 465 466 |
|
ZeusGPULibraryNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Library Not Found Error.
Source code in zeus/device/gpu/common.py
469 470 471 472 473 474 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
472 473 474 |
|
ZeusGPUFunctionNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Function Not Found Error.
Source code in zeus/device/gpu/common.py
477 478 479 480 481 482 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
480 481 482 |
|
ZeusGPUCorruptedInfoROMError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Corrupted Info ROM Error.
Source code in zeus/device/gpu/common.py
485 486 487 488 489 490 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
488 489 490 |
|
ZeusGPULostError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Lost GPU Error.
Source code in zeus/device/gpu/common.py
493 494 495 496 497 498 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
496 497 498 |
|
ZeusGPUResetRequiredError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Reset Required Error.
Source code in zeus/device/gpu/common.py
501 502 503 504 505 506 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
504 505 506 |
|
ZeusGPUOperatingSystemError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Operating System Error.
Source code in zeus/device/gpu/common.py
509 510 511 512 513 514 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
512 513 514 |
|
ZeusGPULibRMVersionMismatchError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps LibRM Version Mismatch Error.
Source code in zeus/device/gpu/common.py
517 518 519 520 521 522 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
520 521 522 |
|
ZeusGPUMemoryError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Memory Error.
Source code in zeus/device/gpu/common.py
525 526 527 528 529 530 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
528 529 530 |
|
ZeusGPUUnknownError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Unknown Error.
Source code in zeus/device/gpu/common.py
533 534 535 536 537 538 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
536 537 538 |
|
ZeusGPUHeterogeneousError
Bases: ZeusBaseGPUError
Exception for when GPUs are not homogeneous.
Source code in zeus/device/gpu/common.py
541 542 543 544 545 546 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
544 545 546 |
|