pass accelerator_config to GangPlacementGroupRequest

ryanaoleary · ryanaoleary · commit d8ff7d500c6c · 2026-05-07T11:00:21.000Z
Signed-off-by: Ryan O'Leary &lt;ryanaoleary@google.com&gt;
diff --git a/python/ray/serve/_private/deployment_scheduler.py b/python/ray/serve/_private/deployment_scheduler.py
@@ -746,7 +746,10 @@ def _schedule_replica(
 
             if slice_pg is not None:
                 slice_pg.shutdown()
-            elif placement_group is not None and scheduling_request.placement_group_bundles is not None:
+            elif (
+                placement_group is not None
+                and scheduling_request.placement_group_bundles is not None
+            ):
                 ray.util.remove_placement_group(placement_group)
 
             return False
diff --git a/python/ray/serve/_private/deployment_state.py b/python/ray/serve/_private/deployment_state.py
@@ -6010,6 +6010,7 @@ def _reserve_gang_placement_groups(
                 replica_pg_fallback_strategy=(
                     replica_config.placement_group_fallback_strategy
                 ),
+                accelerator_config=deployment_state._target_state.info.deployment_config.accelerator_config,
             )
 
         if not gang_requests:

Original file line number	Diff line number	Diff line change
`@@ -6010,6 +6010,7 @@ def _reserve_gang_placement_groups(`
`6010`	`6010`	`replica_pg_fallback_strategy=(`
`6011`	`6011`	`replica_config.placement_group_fallback_strategy`
`6012`	`6012`	`),`
	`6013`	`+ accelerator_config=deployment_state._target_state.info.deployment_config.accelerator_config,`
`6013`	`6014`	`)`
`6014`	`6015`
`6015`	`6016`	`if not gang_requests:`