execute batch when restart session

lablup · Jul 10, 2024 · 0d4fade · 0d4fade
1 parent 52a0d7a
commit 0d4fade
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/ai/backend/manager/registry.py b/src/ai/backend/manager/registry.py
@@ -2691,6 +2691,9 @@ async def _restart_kernel(kernel: KernelRow) -> None:
             SessionStartedEvent(session.id, session.creation_id),
         )
 
+        if session.session_type == SessionTypes.BATCH:
+            await self.trigger_batch_execution(session)
+
     async def execute(
         self,
         session: SessionRow,