not handling gpu mem usage correctly #11

main
Rhys Bailey 2024-06-15 00:22:58 +10:00
parent 2d3182c9be
commit 582d8e5fd7
1 changed files with 1 additions and 1 deletions

View File

@ -231,7 +231,7 @@ class CustomCollector(object):
try: try:
for gpu_name, gpu_stats in stats['gpu_usages'].items(): for gpu_name, gpu_stats in stats['gpu_usages'].items():
add_metric(gpu_usages, [gpu_name], gpu_stats, 'gpu') add_metric(gpu_usages, [gpu_name], gpu_stats, 'gpu')
add_metric(gpu_usages, [gpu_name], gpu_stats, 'mem') add_metric(gpu_mem_usages, [gpu_name], gpu_stats, 'mem')
except KeyError: except KeyError:
pass pass