From 6636583e518dab22269aa73c4953b4004abe1973 Mon Sep 17 00:00:00 2001 From: specture724 Date: Wed, 12 Nov 2025 05:01:47 +0000 Subject: [PATCH] fix: _get_physical_gpu_id requires device_manager --- tests/test_update.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_update.py b/tests/test_update.py index 2c99256..3e173bb 100644 --- a/tests/test_update.py +++ b/tests/test_update.py @@ -68,8 +68,8 @@ def run(): world_size = int(os.getenv("WORLD_SIZE")) ctx = get_context("spawn") queue = ctx.Queue() - _device_uuid = _get_physical_gpu_id(rank) ps = ParameterServer(auto_pg=True) + _device_uuid = _get_physical_gpu_id(ps.device_manager, rank) named_tensors = dict(gen_test_tensors(rank)) checkpoint_name = "test" proc = ctx.Process(target=checker_proc, args=(rank, _device_uuid, named_tensors, queue))