Skip to content

Commit ba70406

Browse files
modifying the nccl manifest URLs to point to the latest one
1 parent 1b83ad5 commit ba70406

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,9 @@ locals {
2626
"a3-highgpu-8g" = {
2727
# Manifest to be installed for enabling TCPX on a3-highgpu-8g machines
2828
gpu_direct_manifests = [
29-
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/fee883360a660f71ba07478db95d5c1325322f77/gpudirect-tcpx/nccl-tcpx-installer.yaml", # nccl_plugin v3.1.9 for tcpx
30-
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/fee883360a660f71ba07478db95d5c1325322f77/gpudirect-tcpx/nccl-config.yaml", # nccl_configmap
31-
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/fee883360a660f71ba07478db95d5c1325322f77/nri_device_injector/nri-device-injector.yaml", # nri_plugin
29+
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/gpudirect-tcpx/nccl-tcpx-installer.yaml", # nccl_plugin v3.1.9 for tcpx
30+
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/gpudirect-tcpx/nccl-config.yaml", # nccl_configmap
31+
"https://raw.githubusercontent.com/GoogleCloudPlatform/container-engine-accelerators/master/nri_device_injector/nri-device-injector.yaml", # nri_plugin
3232
]
3333
updated_workload_path = replace(local.workload_path_tcpx, ".yaml", "-tcpx.yaml")
3434
rxdm_version = "v2.0.12" # matching nccl-tcpx-installer version v3.1.9

0 commit comments

Comments
 (0)