feat: add a FastSocket API

Google APIs · copybara-github · commit 6485d9c71aa7 · 2022-11-28T10:28:58.000-08:00
PiperOrigin-RevId: 491381130
diff --git a/google/container/v1beta1/cluster_service.proto b/google/container/v1beta1/cluster_service.proto
@@ -753,6 +753,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;
@@ -1478,9 +1481,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;
 }
 
@@ -2548,6 +2550,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;
 
@@ -4759,6 +4764,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;
+}
+
 // GetOpenIDConfigRequest gets the OIDC discovery document for the
 // cluster. See the OpenID Connect Discovery 1.0 specification for details.
 message GetOpenIDConfigRequest {
