Skip to content
Snippets Groups Projects
Commit f65d0f9c authored by Nicolas Richart's avatar Nicolas Richart
Browse files

nccl config in modules

parent 31a5b808
No related branches found
No related tags found
1 merge request!3Feat/stack pinot noir
Hide whitespace changes
Inline Side-by-side
stacks/pinot-noir/modules_kuma_h100.yaml
+ 5
− 5
View file @ f65d0f9c
...@@ -7,8 +7,8 @@ modules: ...@@ -7,8 +7,8 @@ modules:
OMPI_MCA_pml: 'ucx' OMPI_MCA_pml: 'ucx'
OMPI_MCA_osc: 'ucx' OMPI_MCA_osc: 'ucx'
UCX_NET_DEVICES: 'mlx5_2:1,mlx5_3:1' UCX_NET_DEVICES: 'mlx5_2:1,mlx5_3:1'
nccl: nccl:
environment: environment:
set: set:
NCCL_IB_HCA: 'mlx5_2:1,mlx5_3:1' NCCL_IB_HCA: 'mlx5_2:1,mlx5_3:1'
NCCL_SOCKET_IFNAME: 'bond0' NCCL_SOCKET_IFNAME: 'bond0'
stacks/pinot-noir/modules_kuma_l40s.yaml
+ 5
− 0
View file @ f65d0f9c
...@@ -7,3 +7,8 @@ modules: ...@@ -7,3 +7,8 @@ modules:
OMPI_MCA_pml: 'ucx' OMPI_MCA_pml: 'ucx'
OMPI_MCA_osc: 'ucx' OMPI_MCA_osc: 'ucx'
UCX_NET_DEVICES: 'mlx5_0:1,mlx5_1:1' UCX_NET_DEVICES: 'mlx5_0:1,mlx5_1:1'
nccl:
environment:
set:
NCCL_IB_HCA: 'mlx5_0:1,mlx5_1:1'
NCCL_SOCKET_IFNAME: 'bond0'
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment