Skip to content

Commit d242b10

Browse files
committed
update nccl-tests.yaml paths
1 parent 59a0c1b commit d242b10

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

micro-benchmarks/nccl-tests/kubernetes/nccl-tests.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ spec:
2121
- name: PATH
2222
value: $PATH:/opt/amazon/efa/bin:/usr/bin
2323
- name: LD_LIBRARY_PATH
24-
value: /opt/amazon/openmpi/lib:/opt/nccl/build/lib:/opt/amazon/efa/lib:/opt/aws-ofi-nccl/install/lib:/usr/local/nvidia/lib:$LD_LIBRARY_PATH
24+
value: /opt/amazon/openmpi/lib:/opt/nccl/build/lib:/opt/amazon/efa/lib:/opt/amazon/ofi-nccl/lib/x86_64-linux-gnu:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
2525
command:
2626
- /opt/amazon/openmpi/bin/mpirun
2727
- --allow-run-as-root
@@ -47,7 +47,7 @@ spec:
4747
- -x
4848
- NCCL_P2P_NET_CHUNKSIZE=524288
4949
- -x
50-
- NCCL_TUNER_PLUGIN=/opt/aws-ofi-nccl/install/lib/libnccl-ofi-tuner.so
50+
- NCCL_TUNER_PLUGIN=/opt/amazon/ofi-nccl/lib/x86_64-linux-gnu/libnccl-ofi-tuner.so
5151
- --mca
5252
- pml
5353
- ^ucx

0 commit comments

Comments
 (0)