File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -160,14 +160,19 @@ ADD nccl-sanity-check.cu /opt
160160RUN install-nccl-sanity-check.sh
161161ADD jax-nccl-test parallel-launch /usr/local/bin/
162162
163+ ###############################################################################
164+ ## Add the systemcheck to the entrypoint.
165+ ###############################################################################
166+
167+ COPY check-shm.sh /opt/nvidia/entrypoint.d/
168+
163169###############################################################################
164170## Install the nsys-jax JAX/XLA-aware profiling scripts, patch Nsight Systems
165171###############################################################################
166172
167173ADD install-nsys-jax.sh /usr/local/bin
168174RUN install-nsys-jax.sh ${JAX_TOOLBOX_REF}
169175
170-
171176###############################################################################
172177## Copy manifest file to the container
173178###############################################################################
You can’t perform that action at this time.
0 commit comments