diff --git a/examples/device/ep/tests/elastic/elastic.py b/examples/device/ep/tests/elastic/elastic.py index 5a8daf9175..50360cac9f 100644 --- a/examples/device/ep/tests/elastic/elastic.py +++ b/examples/device/ep/tests/elastic/elastic.py @@ -1,5 +1,5 @@ # SPDX-FileCopyrightText: Copyright (c) 2025 DeepSeek -# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # This file incorporates material from the DeepSeek project, licensed under the MIT License. # The modifications made by NVIDIA are licensed under the Apache License, Version 2.0. @@ -466,20 +466,6 @@ def worker(torch_rank: int, args: argparse.Namespace): torch.set_default_device("cuda") torch.cuda.set_device(0) - # Initialize UCX - pxb_nics = [ - "mlx5_0", - "mlx5_3", - "mlx5_4", - "mlx5_5", - "mlx5_6", - "mlx5_9", - "mlx5_10", - "mlx5_11", - ] - tcp_nics = ",ibp154s0,ibp192s0,ibp206s0,ibp220s0,ibp94s0" - os.environ["UCX_NET_DEVICES"] = f"cuda0-{pxb_nics[local_rank]}:1" + tcp_nics - # Initialize NIXL os.environ["NIXL_ETCD_ENDPOINTS"] = args.etcd_server