Skip to content

Commit

Permalink
fix: Do not accumulate session occupying slots
Browse files Browse the repository at this point in the history
  • Loading branch information
fregataa committed Dec 2, 2024
1 parent eadddf6 commit cd2f765
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/ai/backend/manager/models/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -1309,7 +1309,7 @@ async def _get_and_transit(
transited = session_row.determine_and_set_status(status_changed_at=now)

def _calculate_session_occupied_slots(session_row: SessionRow):
session_occupying_slots = ResourceSlot.from_json({**session_row.occupying_slots})
session_occupying_slots = ResourceSlot()
for row in session_row.kernels:
kernel_row = cast(KernelRow, row)
kernel_allocs = kernel_row.occupied_slots
Expand Down

0 comments on commit cd2f765

Please sign in to comment.