Update A3 Mega solution to use NCCL plugin 1.0.2

GoogleCloudPlatform · May 31, 2024 · 8e41adf · 8e41adf
1 parent e9614ad
commit 8e41adf
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tools/prologs-epilogs/receive-data-path-manager-mega b/tools/prologs-epilogs/receive-data-path-manager-mega
@@ -30,10 +30,7 @@ fi
 # ensure that dmabuf-import-helper is loaded
 modprobe import-helper
 
-# Populate /etc/hostname for pyxis/enroot
-hostname | tee /etc/hostname
-
-NCCL_PLUGIN_IMAGE=us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/nccl-plugin-gpudirecttcpx-dev:v1.0.1
+NCCL_PLUGIN_IMAGE=us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/nccl-plugin-gpudirecttcpx-dev:v1.0.2
 RXDM_IMAGE=us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/tcpgpudmarxd-dev:v1.0.8
 RXDM_CONTAINER=receive-datapath-manager-"${SLURM_JOB_ID}"
 if [[ ${SLURM_SCRIPT_CONTEXT} == "prolog_slurmd" ]]; then