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]: