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

style: style libnvml

Browse source
This commit is contained in:
Xuehai Pan 2023-08-13 19:32:33 +08:00
parent 5184b7c9a7
commit 643393be53
1 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

21
nvitop/api/libnvml.py
View file

@ -88,7 +88,7 @@ _errcode_to_string = NVMLError._errcode_to_string # pylint: disable=protected-a
# 1. Put error classes in `__all__` first
for _name, _attr in _vars_pynvml.items():
if _name in ('nvmlInit', 'nvmlInitWithFlags', 'nvmlShutdown'):
if _name in {'nvmlInit', 'nvmlInitWithFlags', 'nvmlShutdown'}:
continue
if _name.startswith(('NVML_ERROR_', 'NVMLError_')):
__all__.append(_name)
@ -98,7 +98,7 @@ for _name, _attr in _vars_pynvml.items():
# 2. Then the remaining members
for _name, _attr in _vars_pynvml.items():
if _name in ('nvmlInit', 'nvmlInitWithFlags', 'nvmlShutdown'):
if _name in {'nvmlInit', 'nvmlInitWithFlags', 'nvmlShutdown'}:
continue
if (_name.startswith('NVML_') and not _name.startswith('NVML_ERROR_')) or (
_name.startswith('nvml') and isinstance(_attr, _FunctionType)
@ -171,7 +171,6 @@ del (
_sphinx_doc,
)
# 5. Add explicit references to appease linters
# pylint: disable=no-member
c_nvmlDevice_t: _TypeAlias = _pynvml.c_nvmlDevice_t
@ -780,12 +779,12 @@ if not _pynvml_installation_corrupted:
class c_nvmlMemory_v1_t(_pynvml._PrintableStructure): # pylint: disable=protected-access
_fields_: _ClassVar[list[tuple[str, type]]] = [
# Total physical device memory (in bytes).
('total', _pynvml.c_ulonglong),
('total', _ctypes.c_ulonglong),
# Unallocated device memory (in bytes).
('free', _pynvml.c_ulonglong),
('free', _ctypes.c_ulonglong),
# Allocated device memory (in bytes).
# Note that the driver/GPU always sets aside a small amount of memory for bookkeeping.
('used', _pynvml.c_ulonglong),
('used', _ctypes.c_ulonglong),
]
_fmt_: _ClassVar[dict[str, str]] = {'<default>': '%d B'}
@ -793,16 +792,16 @@ if not _pynvml_installation_corrupted:
class c_nvmlMemory_v2_t(_pynvml._PrintableStructure): # pylint: disable=protected-access
_fields_: _ClassVar[list[tuple[str, type]]] = [
# Structure format version (must be 2).
('version', _pynvml.c_uint),
('version', _ctypes.c_uint),
# Total physical device memory (in bytes).
('total', _pynvml.c_ulonglong),
('total', _ctypes.c_ulonglong),
# Device memory (in bytes) reserved for system use (driver or firmware).
('reserved', _pynvml.c_ulonglong),
('reserved', _ctypes.c_ulonglong),
# Unallocated device memory (in bytes).
('free', _pynvml.c_ulonglong),
('free', _ctypes.c_ulonglong),
# Allocated device memory (in bytes).
# Note that the driver/GPU always sets aside a small amount of memory for bookkeeping.
('used', _pynvml.c_ulonglong),
('used', _ctypes.c_ulonglong),
]
_fmt_: _ClassVar[dict[str, str]] = {'<default>': '%d B'}