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

docs(core/libcuda): update class docstrings

Browse source 
Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-07-22 19:19:29 +08:00
parent 2a519edc54
commit 6533954896
1 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
nvitop/core/libcuda.py
View file

@ -288,11 +288,20 @@ def _extract_cuda_errors_as_classes() -> None:
return new
# pylint: disable=protected-access
new_error_class = type(class_name, (CUDAError,), {'__new__': gen_new(err_val)})
new_error_class.__module__ = __name__
new_error_class.__doc__ = 'CUDA Error with code :data:`{}` ({})'.format(err_name, err_val)
if err_val in CUDAError._errcode_to_string:
new_error_class.__doc__ = 'CUDA Error: {} Code: :data:`{}` ({}).'.format(
CUDAError._errcode_to_string[err_val],
err_name,
err_val,
)
else:
new_error_class.__doc__ = 'CUDA Error with code :data:`{}` ({})'.format(
err_name, err_val
)
setattr(this_module, class_name, new_error_class)
# pylint: disable=protected-access
CUDAError._value_class_mapping[err_val] = new_error_class
CUDAError._errcode_to_name[err_val] = err_name