github-starred/nvitop
2
0
Fork 0
mirror of https://github.com/XuehaiPan/nvitop.git synced 2026-05-21 06:45:24 -06:00
Code Issues 29 Projects Packages Wiki Activity Actions

fix(api/libnvml): fix upstream changes for process info v3 APIs on 535.104.05 driver

Browse source
This commit is contained in:
Xuehai Pan 2023-08-26 10:35:18 +00:00
parent 9f3ff53425
commit 6ac518a775
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
nvitop/api/libnvml.py
View file

@ -606,9 +606,15 @@ if not _pynvml_installation_corrupted:
if lookup('nvmlDeviceGetConfComputeMemSizeInfo'):
if lookup('nvmlDeviceGetComputeRunningProcesses_v3'):
LOGGER.debug(
'NVML get running process version 3 API with v3 type struct is available.',
)
if lookup('nvmlDeviceGetRunningProcessDetailList'):
c_nvmlProcessInfo_t = c_nvmlProcessInfo_v2_t
LOGGER.debug(
'NVML get running process version 3 API with v2 type struct is available.',
)
else:
LOGGER.debug(
'NVML get running process version 3 API with v3 type struct is available.',
)
else:
c_nvmlProcessInfo_t = c_nvmlProcessInfo_v2_t
__get_running_processes_version_suffix = '_v2'