From d46e7b685cc127b79f225380ab00e4b0fd595242 Mon Sep 17 00:00:00 2001 From: Tom Downes Date: Fri, 8 Nov 2024 16:07:40 +0000 Subject: [PATCH] Align GKE documentation and examples with TPG 6.x This commit fixes the documentation and examples to align with changes introduced in a9c2a69f to make gke-node-pool module compatible with TPG 6.x. --- examples/README.md | 4 ++-- modules/compute/gke-node-pool/README.md | 6 +++--- tools/cloud-build/daily-tests/blueprints/ml-gke-e2e.yaml | 8 ++++---- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/examples/README.md b/examples/README.md index 7933f73640..53a84d1a08 100644 --- a/examples/README.md +++ b/examples/README.md @@ -1463,10 +1463,10 @@ guest_accelerator: - type: nvidia-l4 count: 1 gpu_sharing_config: - - max_shared_clients_per_gpu: 2 + max_shared_clients_per_gpu: 2 gpu_sharing_strategy: "TIME_SHARING" gpu_driver_installation_config: - - gpu_driver_version: "LATEST" + gpu_driver_version: "LATEST" ``` * Configuration of the cluster using default drivers provided by GKE. diff --git a/modules/compute/gke-node-pool/README.md b/modules/compute/gke-node-pool/README.md index 40df09bb1e..38ebbedde5 100644 --- a/modules/compute/gke-node-pool/README.md +++ b/modules/compute/gke-node-pool/README.md @@ -151,7 +151,7 @@ The following is an example of guest_accelerator: - gpu_partition_size: 1g.5gb gpu_sharing_config: - - gpu_sharing_strategy: TIME_SHARING + gpu_sharing_strategy: TIME_SHARING max_shared_clients_per_gpu: 3 ``` @@ -181,9 +181,9 @@ The following is an example of using a GPU (with sharing config) attached to an - type: nvidia-tesla-t4 count: 2 gpu_driver_installation_config: - - gpu_driver_version: "LATEST" + gpu_driver_version: "LATEST" gpu_sharing_config: - - max_shared_clients_per_gpu: 2 + max_shared_clients_per_gpu: 2 gpu_sharing_strategy: "TIME_SHARING" ``` diff --git a/tools/cloud-build/daily-tests/blueprints/ml-gke-e2e.yaml b/tools/cloud-build/daily-tests/blueprints/ml-gke-e2e.yaml index d7be384115..83ceb58c65 100644 --- a/tools/cloud-build/daily-tests/blueprints/ml-gke-e2e.yaml +++ b/tools/cloud-build/daily-tests/blueprints/ml-gke-e2e.yaml @@ -70,9 +70,9 @@ deployment_groups: machine_type: g2-standard-4 guest_accelerator: - gpu_driver_installation_config: - - gpu_driver_version: "LATEST" + gpu_driver_version: "LATEST" gpu_sharing_config: - - max_shared_clients_per_gpu: 2 + max_shared_clients_per_gpu: 2 gpu_sharing_strategy: "MPS" - id: job_template_g2_latest_driver @@ -131,9 +131,9 @@ deployment_groups: - type: nvidia-tesla-t4 count: 2 gpu_driver_installation_config: - - gpu_driver_version: "LATEST" + gpu_driver_version: "LATEST" gpu_sharing_config: - - max_shared_clients_per_gpu: 2 + max_shared_clients_per_gpu: 2 gpu_sharing_strategy: "TIME_SHARING" - id: job_template_n1_pool_full_spec