Skip to content

Commit 6661d8a

Browse files
Update rxdm_version info for a3-megagpu-8g machines (GoogleCloudPlatform#4907)
2 parents 689bb9d + be9bc90 commit 6661d8a

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

modules/compute/gke-node-pool/gpu-direct-workload/sample-tcpxo-workload-job.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ spec:
2626
subdomain: nccl-host-1
2727
containers:
2828
- name: nccl-test
29-
image: us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/nccl-plugin-gpudirecttcpx-dev:v1.0.8-1
29+
image: us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/nccl-plugin-gpudirecttcpx-dev:v1.0.14
3030
imagePullPolicy: Always
3131
command:
3232
- /bin/sh

modules/compute/gke-node-pool/gpu_direct.tf

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,11 @@ locals {
4343
"a3-megagpu-8g" = {
4444
# Manifest to be installed for enabling TCPXO on a3-megagpu-8g machines
4545
gpu_direct_manifests = [
46-
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/bd4a7491672b48dfec28f3679b679a614f6cbbc7/gpudirect-tcpxo/nccl-tcpxo-installer.yaml", # nccl_plugin v1.0.8-1 for tcpxo
46+
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/bd4a7491672b48dfec28f3679b679a614f6cbbc7/gpudirect-tcpxo/nccl-tcpxo-installer.yaml", # nccl_plugin v1.0.14 for tcpxo
4747
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/bd4a7491672b48dfec28f3679b679a614f6cbbc7/nri_device_injector/nri-device-injector.yaml", # nri_plugin
4848
]
4949
updated_workload_path = replace(local.workload_path_tcpxo, ".yaml", "-tcpxo.yaml")
50-
rxdm_version = "v1.0.14" # matching nccl-tcpxo-installer version v1.0.8-1
50+
rxdm_version = "v1.0.20" # matching nccl-tcpxo-installer version v1.0.14
5151
min_additional_networks = 8
5252
major_minor_version_acceptable_map = {
5353
"1.28" = "1.28.9-gke.1250000"

0 commit comments

Comments
 (0)