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 |
|
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 |
|
supportsGetTotalEnergyConsumption
abstractmethod
supportsGetTotalEnergyConsumption()
Check if the GPU supports retrieving total energy consumption.
Source code in zeus/device/gpu/common.py
104 105 106 107 |
|
getTotalEnergyConsumption
abstractmethod
getTotalEnergyConsumption()
Return the total energy consumption of the GPU since driver load. Units: mJ.
Source code in zeus/device/gpu/common.py
109 110 111 112 |
|
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
115 116 117 118 119 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 |
|
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
126 127 128 129 |
|
__del__
abstractmethod
__del__()
Shut down the GPU monitoring library to release resources and clean up.
Source code in zeus/device/gpu/common.py
131 132 133 134 |
|
__len__
__len__()
Return the number of GPUs being tracked.
Source code in zeus/device/gpu/common.py
142 143 144 |
|
_ensure_homogeneous
_ensure_homogeneous()
Ensures that all tracked GPUs are homogeneous in terms of name.
Source code in zeus/device/gpu/common.py
146 147 148 149 150 151 |
|
_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
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 |
|
getName
getName(gpu_index)
Return the name of the specified GPU.
Source code in zeus/device/gpu/common.py
171 172 173 |
|
getPowerManagementLimitConstraints
getPowerManagementLimitConstraints(gpu_index)
Return the minimum and maximum power management limits. Units: mW.
Source code in zeus/device/gpu/common.py
175 176 177 |
|
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
179 180 181 182 183 184 |
|
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
186 187 188 189 |
|
setPersistenceMode
setPersistenceMode(gpu_index, enabled, _block=True)
Set persistence mode for the specified GPU.
Source code in zeus/device/gpu/common.py
191 192 193 194 195 196 |
|
getSupportedMemoryClocks
getSupportedMemoryClocks(gpu_index)
Return a list of supported memory clock frequencies. Units: MHz.
Source code in zeus/device/gpu/common.py
198 199 200 |
|
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
202 203 204 205 206 207 208 209 210 211 |
|
resetMemoryLockedClocks
resetMemoryLockedClocks(gpu_index, _block=True)
Reset the locked memory clocks to the default.
Source code in zeus/device/gpu/common.py
213 214 215 216 |
|
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
218 219 220 221 222 223 224 225 226 227 228 229 |
|
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
231 232 233 234 235 236 237 238 239 240 |
|
resetGpuLockedClocks
resetGpuLockedClocks(gpu_index, _block=True)
Reset the locked GPU clocks to the default.
Source code in zeus/device/gpu/common.py
242 243 244 245 |
|
getInstantPowerUsage
getInstantPowerUsage(gpu_index)
Return the current power draw of the GPU. Units: mW.
Source code in zeus/device/gpu/common.py
247 248 249 |
|
supportsGetTotalEnergyConsumption
supportsGetTotalEnergyConsumption(gpu_index)
Check if the GPU supports retrieving total energy consumption.
Source code in zeus/device/gpu/common.py
251 252 253 |
|
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
255 256 257 |
|
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
260 261 262 263 264 265 266 267 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 |
|
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
267 268 269 270 271 272 |
|
__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
274 275 276 277 278 279 |
|
__len__
__len__()
Return 0, indicating no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
286 287 288 |
|
_ensure_homogeneous
_ensure_homogeneous()
Raise a ValueError as no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
290 291 292 |
|
_warn_sys_admin
_warn_sys_admin()
Raise a ValueError as no GPUs are being tracked.
Source code in zeus/device/gpu/common.py
294 295 296 |
|
getName
getName(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
298 299 300 |
|
getPowerManagementLimitConstraints
getPowerManagementLimitConstraints(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
302 303 304 |
|
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
306 307 308 309 310 |
|
resetPowerManagementLimit
resetPowerManagementLimit(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
312 313 314 |
|
setPersistenceMode
setPersistenceMode(gpu_index, enabled, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
316 317 318 319 320 |
|
getSupportedMemoryClocks
getSupportedMemoryClocks(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
322 323 324 |
|
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
326 327 328 329 330 331 332 333 334 |
|
resetMemoryLockedClocks
resetMemoryLockedClocks(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
336 337 338 |
|
getSupportedGraphicsClocks
getSupportedGraphicsClocks(
gpu_index, memory_clock_mhz=None
)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
340 341 342 343 344 |
|
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
346 347 348 349 350 351 352 353 354 |
|
resetGpuLockedClocks
resetGpuLockedClocks(gpu_index, _block=True)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
356 357 358 |
|
getInstantPowerUsage
getInstantPowerUsage(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
360 361 362 |
|
supportsGetTotalEnergyConsumption
supportsGetTotalEnergyConsumption(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
364 365 366 |
|
getTotalEnergyConsumption
getTotalEnergyConsumption(gpu_index)
Raise a ValueError as no GPUs are available.
Source code in zeus/device/gpu/common.py
368 369 370 |
|
ZeusGPUInitError
Bases: ZeusBaseGPUError
Import error or GPU library initialization failures.
Source code in zeus/device/gpu/common.py
373 374 375 376 377 378 |
|
__init__
__init__(message)
Source code in zeus/device/gpu/common.py
376 377 378 |
|
ZeusGPUInvalidArgError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Invalid Argument.
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 |
|
ZeusGPUNotSupportedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Not Supported Operation on GPU.
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 |
|
ZeusGPUNoPermissionError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps No Permission to perform GPU operation.
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 |
|
ZeusGPUAlreadyInitializedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Already Initialized GPU.
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 |
|
ZeusGPUNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Not Found 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 |
|
ZeusGPUInsufficientSizeError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Size.
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 |
|
ZeusGPUInsufficientPowerError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Power.
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 |
|
ZeusGPUDriverNotLoadedError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Driver Error.
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 |
|
ZeusGPUTimeoutError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Timeout 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 |
|
ZeusGPUIRQError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps IRQ 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 |
|
ZeusGPULibraryNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Library Not Found 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 |
|
ZeusGPUFunctionNotFoundError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Function 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 |
|
ZeusGPUCorruptedInfoROMError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Corrupted Info ROM 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 |
|
ZeusGPULostError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Lost GPU 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 |
|
ZeusGPUResetRequiredError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Reset Required 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 |
|
ZeusGPUOperatingSystemError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Operating System 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 |
|
ZeusGPULibRMVersionMismatchError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps LibRM Version Mismatch 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 |
|
ZeusGPUMemoryError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Insufficient Memory 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 |
|
ZeusGPUUnknownError
Bases: ZeusBaseGPUError
Zeus GPU exception that wraps Unknown 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 |
|
ZeusGPUHeterogeneousError
Bases: ZeusBaseGPUError
Exception for when GPUs are not homogeneous.
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 |
|