Skip to content

Commit 30e590e

Browse files
committed
remove redundant CI case
1 parent 2c611c9 commit 30e590e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

verl/single_controller/base/worker.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,7 @@ def _setup_env_cuda_visible_devices(self):
242242
# so we need to set local rank when the flag is set.
243243
device_name = "NPU" if is_npu_available else "GPU"
244244
local_rank = ray.get_runtime_context().get_accelerator_ids()[device_name][0]
245+
local_rank = os.environ.get("RAY_LOCAL_RANK")
245246
os.environ["LOCAL_RANK"] = local_rank
246247
get_torch_device().set_device(int(local_rank))
247248

0 commit comments

Comments
 (0)