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

feat(api/process): update GPU status for gone processes

Browse source
This commit is contained in:
Xuehai Pan 2023-08-04 15:47:55 +08:00
parent e41e0ea701
commit b99d90544d
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
nvitop/api/process.py
View file

@ -694,7 +694,19 @@ class GpuProcess: # pylint: disable=too-many-instance-attributes,too-many-publi
self.set_gpu_memory(NA)
self.set_gpu_cc_protected_memory(NA)
self.set_gpu_utilization(NA, NA, NA, NA)
self.device.processes()
processes = self.device.processes()
process = processes.get(self.pid, self)
if process is not self:
# The current process is gone and the instance has been removed from the cache.
# Update GPU status from the new instance.
self.set_gpu_memory(process.gpu_memory())
self.set_gpu_cc_protected_memory(process.gpu_cc_protected_memory())
self.set_gpu_utilization(
process.gpu_sm_utilization(),
process.gpu_memory_utilization(),
process.gpu_encoder_utilization(),
process.gpu_decoder_utilization(),
)
return self.gpu_memory()
@property