Skip to content

Commit

Permalink
Fix name attribute of GPU
Browse files Browse the repository at this point in the history
  • Loading branch information
bkbilly committed Oct 10, 2023
1 parent d4cbae5 commit 7f3b716
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lnxlink/modules/gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def get_info(self):
for gpu_id in range(self.gpu_ids["amd"]):
amd_gpu = pyamdgpuinfo.get_gpu(gpu_id)
gpus[f"amd_{gpu_id}"] = {
"name": amd_gpu.name,
"Name": amd_gpu.name,
"VRAM usage": amd_gpu.query_vram_usage(),
"GTT usage": amd_gpu.query_gtt_usage(),
"Temperature": amd_gpu.query_temperature(),
Expand All @@ -45,7 +45,7 @@ def get_info(self):
gpu_util = nvidia_gpu.gpu_util
gpu_util = self._older_gpu_load(gpu_id, gpu_util)
gpus[f"nvidia_{gpu_id}"] = {
"name": nvidia_gpu.name,
"Name": nvidia_gpu.name,
"Memory usage": round(nvidia_gpu.mem_util, 1),
"load": gpu_util,
"Temperature": nvidia_gpu.temperature,
Expand Down

0 comments on commit 7f3b716

Please sign in to comment.