Skip to content

Commit b59e57e

Browse files
fix division by max_gpu
1 parent eb5ea04 commit b59e57e

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

nodes/manager/manager.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,9 +172,9 @@ def get_load(self):
172172
active_jobs_info = self.get_active_jobs_info()
173173
queued_jobs_info = self.get_queued_jobs_info()["queued_jobs"]
174174
sum_gpu_mem = (
175-
sum(self.gpu_devices_mem_available)
176-
if isinstance(self.gpu_devices_mem_available, list)
177-
else self.gpu_devices_mem_available
175+
sum(self.gpu_devices_mem_max)
176+
if isinstance(self.gpu_devices_mem_max, list)
177+
else self.gpu_devices_mem_max
178178
)
179179
tot_required_gpu_mem = sum(
180180
x["required_gpu_mem"] for x in active_jobs_info + queued_jobs_info

0 commit comments

Comments
 (0)