feat: add AcceleratorConfig to ListAcceleratorTypesResponse

Google APIs · copybara-github · commit 08882cfb3fb4 · 2023-03-07T16:32:59.000-08:00
PiperOrigin-RevId: 514873293
diff --git a/google/cloud/tpu/v2/cloud_tpu.proto b/google/cloud/tpu/v2/cloud_tpu.proto
@@ -460,6 +460,9 @@ message Node {
 
   // Shielded Instance options.
   ShieldedInstanceConfig shielded_instance_config = 45;
+
+  // The AccleratorConfig for the TPU Node.
+  AcceleratorConfig accelerator_config = 46;
 }
 
 // Request for [ListNodes][google.cloud.tpu.v2.Tpu.ListNodes].
@@ -591,6 +594,9 @@ message AcceleratorType {
 
   // the accelerator type.
   string type = 2;
+
+  // The accelerator config.
+  repeated AcceleratorConfig accelerator_configs = 3;
 }
 
 // Request for [GetAcceleratorType][google.cloud.tpu.v2.Tpu.GetAcceleratorType].
@@ -792,6 +798,30 @@ message GetGuestAttributesResponse {
   repeated GuestAttributes guest_attributes = 1;
 }
 
+// A TPU accelerator configuration.
+message AcceleratorConfig {
+  // TPU type.
+  enum Type {
+    // Unspecified version.
+    TYPE_UNSPECIFIED = 0;
+
+    // TPU v2.
+    V2 = 2;
+
+    // TPU v3.
+    V3 = 4;
+
+    // TPU v4.
+    V4 = 7;
+  }
+
+  // Required. Type of TPU.
+  Type type = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Required. Topology of TPU in chips.
+  string topology = 2 [(google.api.field_behavior) = REQUIRED];
+}
+
 // A set of Shielded Instance options.
 message ShieldedInstanceConfig {
   // Defines whether the instance has Secure Boot enabled.
