From 40167b2c9b430b31a7bbd58db25d2d0109f059bd Mon Sep 17 00:00:00 2001 From: jianjunzhong Date: Tue, 13 Jan 2026 17:31:20 +0800 Subject: [PATCH] [ray] fix: use get_device_name() for automatic device detection instead of by parameter passing Signed-off-by: jianjunzhong --- verl/single_controller/ray/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/verl/single_controller/ray/base.py b/verl/single_controller/ray/base.py index 2f657c08221..95e5947b971 100644 --- a/verl/single_controller/ray/base.py +++ b/verl/single_controller/ray/base.py @@ -377,7 +377,7 @@ def __init__( # if a WorkerGroup is spawned from Colocate WorkerGroup, this indicates which sub-class is binded to # this WorkerGroup. self.sub_cls_name = "" - self.device_name = kwargs.get("device_name", "cuda") + self.device_name = get_device_name() self.profile_steps = kwargs.get("profile_steps", None) self.worker_nsight_options = kwargs.get("worker_nsight_options", None) self.customized_worker_env = kwargs.get("worker_env", {})