add tests

NVIDIA · Aug 20, 2024 · 2f46e8d · 2f46e8d
1 parent 31d2791
commit 2f46e8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/container/Dockerfile.base b/.github/container/Dockerfile.base
@@ -181,6 +181,7 @@ ENV PATH=/opt/amazon/efa/bin:${PATH}
 ADD install-nccl-sanity-check.sh /usr/local/bin
 ADD nccl-sanity-check.cu /opt
 RUN install-nccl-sanity-check.sh
+ADD jax-nccl-test.py jax-nccl-test-multiprocess.sh /usr/local/bin
 
 ###############################################################################
 ## Add the systemcheck to the entrypoint.