From 0632ed8778cab44de6152eb873d09fa40c241962 Mon Sep 17 00:00:00 2001 From: Ryan Rock Date: Fri, 20 Feb 2026 15:33:04 -0600 Subject: [PATCH] [AMD][CI] Fix test_custom_allreduce for A100 testgroup (#34735) Signed-off-by: Ryan Rock --- tests/distributed/test_custom_all_reduce.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/distributed/test_custom_all_reduce.py b/tests/distributed/test_custom_all_reduce.py index f6e274be9..68abc2b98 100644 --- a/tests/distributed/test_custom_all_reduce.py +++ b/tests/distributed/test_custom_all_reduce.py @@ -33,6 +33,7 @@ def graph_allreduce( ): with monkeypatch.context() as m: m.delenv("CUDA_VISIBLE_DEVICES", raising=False) + m.delenv("HIP_VISIBLE_DEVICES", raising=False) device = torch.device(f"cuda:{rank}") torch.cuda.set_device(device) init_test_distributed_environment(tp_size, pp_size, rank, distributed_init_port) @@ -92,6 +93,7 @@ def eager_allreduce( ): with monkeypatch.context() as m: m.delenv("CUDA_VISIBLE_DEVICES", raising=False) + m.delenv("HIP_VISIBLE_DEVICES", raising=False) device = torch.device(f"cuda:{rank}") torch.cuda.set_device(device) init_test_distributed_environment(tp_size, pp_size, rank, distributed_init_port)