feat: add a FastSocket API

Google APIs · copybara-github · commit 39bcdf411c9c · 2022-11-27T23:48:55.000-08:00
PiperOrigin-RevId: 491259489
diff --git a/google/container/v1/cluster_service.proto b/google/container/v1/cluster_service.proto
@@ -759,6 +759,9 @@ message NodeConfig {
   // All the nodes in the node pool will be Confidential VM once enabled.
   ConfidentialNodes confidential_nodes = 35;
 
+  // Enable or disable NCCL fast socket for the node pool.
+  optional FastSocket fast_socket = 36;
+
   // The resource labels for the node pool to use to annotate any related
   // Google Compute Engine resources.
   map<string, string> resource_labels = 37;
@@ -1278,9 +1281,8 @@ message BinaryAuthorization {
   // anything other than EVALUATION_MODE_UNSPECIFIED, this field is ignored.
   bool enabled = 1 [deprecated = true];
 
-  // Mode of operation for binauthz policy evaluation. Currently the only
-  // options are equivalent to enable/disable. If unspecified, defaults to
-  // DISABLED.
+  // Mode of operation for binauthz policy evaluation. If unspecified, defaults
+  // to DISABLED.
   EvaluationMode evaluation_mode = 2;
 }
 
@@ -2284,6 +2286,9 @@ message UpdateNodePoolRequest {
   // Enable or disable gvnic on the node pool.
   VirtualNIC gvnic = 29;
 
+  // Enable or disable NCCL fast socket for the node pool.
+  FastSocket fast_socket = 31;
+
   // Logging configuration.
   NodePoolLoggingConfig logging_config = 32;
 
@@ -4216,6 +4221,12 @@ message VirtualNIC {
   bool enabled = 1;
 }
 
+// Configuration of Fast Socket feature.
+message FastSocket {
+  // Whether Fast Socket features are enabled in the node pool.
+  bool enabled = 1;
+}
+
 // PrivateIPv6GoogleAccess controls whether and how the pods can communicate
 // with Google Services through gRPC over IPv6.
 enum PrivateIPv6GoogleAccess {
