From cb357fbd80959ac95a097cf4b32774bc1d47dda4 Mon Sep 17 00:00:00 2001
From: Klaus Ma <klausm@nvidia.com>
Date: Mon, 22 Jun 2026 17:48:08 +0800
Subject: [PATCH] gang: track incomplete_tasks on-demand via bind/unbind hooks

---
 session_manager/src/scheduler/plugins/gang.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/session_manager/src/scheduler/plugins/gang.rs b/session_manager/src/scheduler/plugins/gang.rs
index abb309b5..7da0fc2d 100644
--- a/session_manager/src/scheduler/plugins/gang.rs
+++ b/session_manager/src/scheduler/plugins/gang.rs
@@ -154,6 +154,7 @@ impl Plugin for GangPlugin {
     fn on_session_bind(&mut self, ssn: SessionInfoPtr) {
         if let Some(state) = self.ssn_state.get_mut(&ssn.id) {
             state.bound += 1;
+            state.incomplete_tasks = state.incomplete_tasks.saturating_sub(1);
         }
     }
 
@@ -166,6 +167,7 @@ impl Plugin for GangPlugin {
     fn on_session_unbind(&mut self, ssn: SessionInfoPtr) {
         if let Some(state) = self.ssn_state.get_mut(&ssn.id) {
             state.bound = state.bound.saturating_sub(1);
+            state.incomplete_tasks = state.incomplete_tasks.saturating_add(1);
         }
     }
 }