Merge pull request GoogleCloudPlatform#4303 from cboneti/nccl-1.0.6

ighosh98 · web-flow · commit e69de3ef3928 · 2025-06-23T19:59:44.000+05:30
update nccl to 1.0.6
diff --git a/examples/gke-a3-ultragpu/gke-a3-ultragpu.yaml b/examples/gke-a3-ultragpu/gke-a3-ultragpu.yaml
@@ -226,7 +226,7 @@ deployment_groups:
         install: true  # NCCL gIB plugin via DaemonSet initContainer
         path: $(vars.gib_installer_path)
         template_vars:
-          version: v1.0.5
+          version: v1.0.6
           accelerator_count: 8
 
   - id: job-template
diff --git a/examples/gke-a3-ultragpu/nccl-jobset-example.yaml b/examples/gke-a3-ultragpu/nccl-jobset-example.yaml
@@ -125,7 +125,7 @@ spec:
             - name: nccl
               stdin: true
               tty: true
-              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
               securityContext:
                 privileged: true
               env:
diff --git a/examples/gke-a3-ultragpu/system_benchmarks/ramble-nccl.yaml b/examples/gke-a3-ultragpu/system_benchmarks/ramble-nccl.yaml
@@ -126,7 +126,7 @@ data:
           | sed 's/ / -x /g')
 
         container_name: nccl-tests
-        container_uri: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+        container_uri: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
         gke_container_name: nccl
         gke_namespace: ramble
         gpus_per_node: 8
diff --git a/examples/gke-a4/gke-a4.yaml b/examples/gke-a4/gke-a4.yaml
@@ -243,7 +243,7 @@ deployment_groups:
         install: true  # NCCL gIB plugin via DaemonSet initContainer
         path: $(vars.gib_installer_path)
         template_vars:
-          version: v1.0.5
+          version: v1.0.6
           accelerator_count: 8
 
   - id: job-template
diff --git a/examples/gke-a4/nccl-jobset-example.yaml b/examples/gke-a4/nccl-jobset-example.yaml
@@ -125,7 +125,7 @@ spec:
             - name: nccl
               stdin: true
               tty: true
-              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
               securityContext:
                 privileged: true
               env:
diff --git a/examples/gke-consumption-options/dws-flex-start-queued-provisioning/nccl-installer.yaml b/examples/gke-consumption-options/dws-flex-start-queued-provisioning/nccl-installer.yaml
@@ -70,7 +70,7 @@ spec:
           sysctl -w net.ipv4.conf.gpu5rdma0.log_martians=0
           sysctl -w net.ipv4.conf.gpu6rdma0.log_martians=0
           sysctl -w net.ipv4.conf.gpu7rdma0.log_martians=0
-      - image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib:v1.0.5
+      - image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib:v1.0.6
         name: nccl-rdma-installer
         resources:
           requests:
diff --git a/examples/gke-consumption-options/dws-flex-start-queued-provisioning/nccl-jobset-example.yaml b/examples/gke-consumption-options/dws-flex-start-queued-provisioning/nccl-jobset-example.yaml
@@ -129,7 +129,7 @@ spec:
             - name: nccl
               stdin: true
               tty: true
-              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
               securityContext:
                 privileged: true
               env:
diff --git a/examples/gke-consumption-options/dws-flex-start/nccl-installer.yaml b/examples/gke-consumption-options/dws-flex-start/nccl-installer.yaml
@@ -70,7 +70,7 @@ spec:
           sysctl -w net.ipv4.conf.gpu5rdma0.log_martians=0
           sysctl -w net.ipv4.conf.gpu6rdma0.log_martians=0
           sysctl -w net.ipv4.conf.gpu7rdma0.log_martians=0
-      - image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib:v1.0.5
+      - image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib:v1.0.6
         name: nccl-rdma-installer
         resources:
           requests:
diff --git a/examples/gke-consumption-options/dws-flex-start/nccl-jobset-example.yaml b/examples/gke-consumption-options/dws-flex-start/nccl-jobset-example.yaml
@@ -126,7 +126,7 @@ spec:
             - name: nccl
               stdin: true
               tty: true
-              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+              image: us-docker.pkg.dev/gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
               securityContext:
                 privileged: true
               env:
diff --git a/examples/hypercompute_clusters/a3u-slurm-ubuntu-gcs/a3u-slurm-ubuntu-gcs.yaml b/examples/hypercompute_clusters/a3u-slurm-ubuntu-gcs/a3u-slurm-ubuntu-gcs.yaml
@@ -48,7 +48,7 @@ vars:
     project: $(vars.project_id)
     family: $(vars.deployment_name)-u22
   disk_size_gb: 200
-  nccl_plugin_version: v1.0.5
+  nccl_plugin_version: v1.0.6
   benchmark_dir: $(ghpc_stage("system_benchmarks"))
 
   # Here we define a set of startup script runners that are used to configure
diff --git a/examples/machine-learning/a3-ultragpu-8g/a3ultra-slurm-blueprint.yaml b/examples/machine-learning/a3-ultragpu-8g/a3ultra-slurm-blueprint.yaml
@@ -39,7 +39,7 @@ vars:
     project: $(vars.project_id)
     family: $(vars.deployment_name)-u22
   disk_size_gb: 100
-  nccl_plugin_version: v1.0.5
+  nccl_plugin_version: v1.0.6
   base_network_name: $(vars.deployment_name)
 
   #Provisioning models (set to true or fill in reservation name, pick only one)
diff --git a/examples/machine-learning/a4-highgpu-8g/a4high-slurm-blueprint.yaml b/examples/machine-learning/a4-highgpu-8g/a4high-slurm-blueprint.yaml
@@ -39,7 +39,7 @@ vars:
     project: $(vars.project_id)
     family: $(vars.deployment_name)-u22
   disk_size_gb: 100
-  nccl_plugin_version: v1.0.5
+  nccl_plugin_version: v1.0.6
   benchmark_dir: $(ghpc_stage("system_benchmarks"))
   base_network_name: $(vars.deployment_name)
 
diff --git a/examples/machine-learning/a4-highgpu-8g/system_benchmarks/run-nccl-tests-via-ramble.sh b/examples/machine-learning/a4-highgpu-8g/system_benchmarks/run-nccl-tests-via-ramble.sh
@@ -97,8 +97,8 @@ ramble:
     hostlist: \${SLURM_JOB_NODELIST}
 
     container_dir: "/opt/apps/ramble/sqsh"
-    container_name: nccl-plugin-gib-diagnostic-v1.0.5
-    container_uri: docker://us-docker.pkg.dev#gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.5
+    container_name: nccl-plugin-gib-diagnostic-v1.0.6
+    container_uri: docker://us-docker.pkg.dev#gce-ai-infra/gpudirect-gib/nccl-plugin-gib-diagnostic:v1.0.6
     processes_per_node: 8
     processes_per_node: '{gpus_per_node}'
     gpus_per_node: '8'
diff --git a/modules/management/kubectl-apply/variables.tf b/modules/management/kubectl-apply/variables.tf
@@ -17,7 +17,7 @@
 locals {
   kueue_supported_versions  = ["v0.11.4", "v0.10.1", "v0.10.0", "v0.9.1", "v0.9.0", "v0.8.1"]
   jobset_supported_versions = ["v0.8.1", "v0.7.2", "v0.5.2"]
-  gib_supported_versions    = ["v1.0.2", "v1.0.3", "v1.0.5"]
+  gib_supported_versions    = ["v1.0.2", "v1.0.3", "v1.0.5", "v1.0.6"]
 }
 
 resource "terraform_data" "kueue_validations" {
