From 6023bcaa2cddffc6c00397224ebba679c4c7eac3 Mon Sep 17 00:00:00 2001 From: Matthias Reso <13337103+mreso@users.noreply.github.com> Date: Thu, 13 Nov 2025 22:01:58 +0000 Subject: [PATCH] Added NCCL_IB_HCA as an env variable --- src/forge/env.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/forge/env.py b/src/forge/env.py index b0d365350..b290de9f8 100644 --- a/src/forge/env.py +++ b/src/forge/env.py @@ -118,6 +118,12 @@ def get_value(self) -> Any: description="Whether or not to use RDMA in TorchStore.", ) +NCCL_IB_HCA = EnvVar( + name="NCCL_IB_HCA", + default="mlx5_1:1,mlx5_2:1,mlx5_3:1,mlx5_4:1", + description="IB interfaces.", +) + def all_env_vars() -> list[EnvVar]: