feat: Add multihost_gpu_node_count to Vertex SDK for multihost GPU support

Google APIs · copybara-github · commit 4129cd809566 · 2025-03-05T09:44:56.000-08:00
PiperOrigin-RevId: 733768886
diff --git a/google/cloud/aiplatform/v1beta1/machine_resources.proto b/google/cloud/aiplatform/v1beta1/machine_resources.proto
@@ -58,6 +58,13 @@ message MachineSpec {
   // available from GKE. (Example: tpu_topology: "2x2x1").
   string tpu_topology = 4 [(google.api.field_behavior) = IMMUTABLE];
 
+  // Optional. Immutable. The number of nodes per replica for multihost GPU
+  // deployments.
+  int32 multihost_gpu_node_count = 6 [
+    (google.api.field_behavior) = IMMUTABLE,
+    (google.api.field_behavior) = OPTIONAL
+  ];
+
   // Optional. Immutable. Configuration controlling how this resource pool
   // consumes reservation.
   ReservationAffinity reservation_affinity = 5 [
