fix: Create kernels with correct scaling group value (#2837)

lablup · Sep 14, 2024 · 97f2390 · 97f2390
1 parent df4ae50
commit 97f2390
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/changes/2837.fix.md b/changes/2837.fix.md
@@ -0,0 +1 @@
+Create kernels with correct `scaling_group` value.
diff --git a/src/ai/backend/manager/registry.py b/src/ai/backend/manager/registry.py
@@ -1332,7 +1332,7 @@ async def start_session(
                 agent_alloc_ctx=AgentAllocationContext(
                     agent_id=k.agent,
                     agent_addr=k.agent_addr,
-                    scaling_group=scheduled_session.scaling_group,
+                    scaling_group=scheduled_session.scaling_group_name,
                 ),
                 allocated_host_ports=set(),
             )
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Create kernels with correct `scaling_group` value.