diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-01-04 00:40:16 -0500 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-01-06 00:14:20 -0500 |
| commit | f6ab5a39d762a7791573d1c52ae5a3024b10e8ed (patch) | |
| tree | c3958d77a6dae42457b571dbe0f1efec7ce45dd2 /modules/memmon.py | |
| parent | d782a95967c9eea753df3333cd1954b6ec73eba0 (diff) | |
| parent | 3e22e294135ed0327ce9d9738655ff03c53df3c0 (diff) | |
Merge branch 'AUTOMATIC1111:master' into sub-quad_attn_opt
Diffstat (limited to 'modules/memmon.py')
| -rw-r--r-- | modules/memmon.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/memmon.py b/modules/memmon.py index 9fb9b687..a7060f58 100644 --- a/modules/memmon.py +++ b/modules/memmon.py @@ -71,10 +71,13 @@ class MemUsageMonitor(threading.Thread): def read(self): if not self.disabled: free, total = torch.cuda.mem_get_info() + self.data["free"] = free self.data["total"] = total torch_stats = torch.cuda.memory_stats(self.device) + self.data["active"] = torch_stats["active.all.current"] self.data["active_peak"] = torch_stats["active_bytes.all.peak"] + self.data["reserved"] = torch_stats["reserved_bytes.all.current"] self.data["reserved_peak"] = torch_stats["reserved_bytes.all.peak"] self.data["system_peak"] = total - self.data["min_free"] |