feat: add TPU_V2 & TPU_V3 values to AcceleratorType in aiplatform v1/v1beta1 accelerator_type.proto

PiperOrigin-RevId: 430259767
3 years ago · f873e7f132
parent 3e5d2eab1a
commit f873e7f132
2 changed files with 12 additions and 4 deletions
--- a/google/cloud/aiplatform/v1/accelerator_type.proto
+++ b/google/cloud/aiplatform/v1/accelerator_type.proto
@ -16,8 +16,6 @@ syntax = "proto3";

 package google.cloud.aiplatform.v1;

-import "google/api/annotations.proto";
-
 option csharp_namespace = "Google.Cloud.AIPlatform.V1";
 option go_package = "google.golang.org/genproto/googleapis/cloud/aiplatform/v1;aiplatform";
 option java_multiple_files = true;
@ -48,4 +46,10 @@ enum AcceleratorType {

  // Nvidia Tesla A100 GPU.
  NVIDIA_TESLA_A100 = 8;
+
+  // TPU v2.
+  TPU_V2 = 6;
+
+  // TPU v3.
+  TPU_V3 = 7;
 }
--- a/google/cloud/aiplatform/v1beta1/accelerator_type.proto
+++ b/google/cloud/aiplatform/v1beta1/accelerator_type.proto
@ -16,8 +16,6 @@ syntax = "proto3";

 package google.cloud.aiplatform.v1beta1;

-import "google/api/annotations.proto";
-
 option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
 option go_package = "google.golang.org/genproto/googleapis/cloud/aiplatform/v1beta1;aiplatform";
 option java_multiple_files = true;
@ -48,4 +46,10 @@ enum AcceleratorType {

  // Nvidia Tesla A100 GPU.
  NVIDIA_TESLA_A100 = 8;
+
+  // TPU v2.
+  TPU_V2 = 6;
+
+  // TPU v3.
+  TPU_V3 = 7;
 }