diff --git a/google/cloud/container_v1beta1/__init__.py b/google/cloud/container_v1beta1/__init__.py index 72475a5a..1993fd79 100644 --- a/google/cloud/container_v1beta1/__init__.py +++ b/google/cloud/container_v1beta1/__init__.py @@ -52,6 +52,7 @@ DnsCacheConfig, DNSConfig, EphemeralStorageConfig, + FastSocket, GatewayAPIConfig, GcePersistentDiskCsiDriverConfig, GcfsConfig, @@ -211,6 +212,7 @@ "DeleteNodePoolRequest", "DnsCacheConfig", "EphemeralStorageConfig", + "FastSocket", "GPUSharingConfig", "GatewayAPIConfig", "GcePersistentDiskCsiDriverConfig", diff --git a/google/cloud/container_v1beta1/types/__init__.py b/google/cloud/container_v1beta1/types/__init__.py index 41669142..c13a4196 100644 --- a/google/cloud/container_v1beta1/types/__init__.py +++ b/google/cloud/container_v1beta1/types/__init__.py @@ -46,6 +46,7 @@ DnsCacheConfig, DNSConfig, EphemeralStorageConfig, + FastSocket, GatewayAPIConfig, GcePersistentDiskCsiDriverConfig, GcfsConfig, @@ -202,6 +203,7 @@ "DnsCacheConfig", "DNSConfig", "EphemeralStorageConfig", + "FastSocket", "GatewayAPIConfig", "GcePersistentDiskCsiDriverConfig", "GcfsConfig", diff --git a/google/cloud/container_v1beta1/types/cluster_service.py b/google/cloud/container_v1beta1/types/cluster_service.py index cd697747..6141a6a7 100644 --- a/google/cloud/container_v1beta1/types/cluster_service.py +++ b/google/cloud/container_v1beta1/types/cluster_service.py @@ -158,6 +158,7 @@ "ResourceUsageExportConfig", "ShieldedNodes", "VirtualNIC", + "FastSocket", "GetOpenIDConfigRequest", "GetOpenIDConfigResponse", "GetJSONWebKeysRequest", @@ -510,6 +511,11 @@ class NodeConfig(proto.Message): Confidential nodes config. All the nodes in the node pool will be Confidential VM once enabled. + fast_socket (google.cloud.container_v1beta1.types.FastSocket): + Enable or disable NCCL fast socket for the + node pool. + + This field is a member of `oneof`_ ``_fast_socket``. resource_labels (MutableMapping[str, str]): The resource labels for the node pool to use to annotate any related Google Compute Engine @@ -645,6 +651,12 @@ class NodeConfig(proto.Message): number=35, message="ConfidentialNodes", ) + fast_socket: "FastSocket" = proto.Field( + proto.MESSAGE, + number=36, + optional=True, + message="FastSocket", + ) resource_labels: MutableMapping[str, str] = proto.MapField( proto.STRING, proto.STRING, @@ -1906,9 +1918,8 @@ class BinaryAuthorization(proto.Message): EVALUATION_MODE_UNSPECIFIED, this field is ignored. evaluation_mode (google.cloud.container_v1beta1.types.BinaryAuthorization.EvaluationMode): Mode of operation for binauthz policy - evaluation. Currently the only options are - equivalent to enable/disable. If unspecified, - defaults to DISABLED. + evaluation. If unspecified, defaults to + DISABLED. """ class EvaluationMode(proto.Enum): @@ -3670,6 +3681,9 @@ class UpdateNodePoolRequest(proto.Message): Confidential VM once enabled. gvnic (google.cloud.container_v1beta1.types.VirtualNIC): Enable or disable gvnic on the node pool. + fast_socket (google.cloud.container_v1beta1.types.FastSocket): + Enable or disable NCCL fast socket for the + node pool. logging_config (google.cloud.container_v1beta1.types.NodePoolLoggingConfig): Logging configuration. resource_labels (google.cloud.container_v1beta1.types.ResourceLabels): @@ -3765,6 +3779,11 @@ class UpdateNodePoolRequest(proto.Message): number=29, message="VirtualNIC", ) + fast_socket: "FastSocket" = proto.Field( + proto.MESSAGE, + number=31, + message="FastSocket", + ) logging_config: "NodePoolLoggingConfig" = proto.Field( proto.MESSAGE, number=32, @@ -7248,6 +7267,21 @@ class VirtualNIC(proto.Message): ) +class FastSocket(proto.Message): + r"""Configuration of Fast Socket feature. + + Attributes: + enabled (bool): + Whether Fast Socket features are enabled in + the node pool. + """ + + enabled: bool = proto.Field( + proto.BOOL, + number=1, + ) + + class GetOpenIDConfigRequest(proto.Message): r"""GetOpenIDConfigRequest gets the OIDC discovery document for the cluster. See the OpenID Connect Discovery 1.0 specification diff --git a/scripts/fixup_container_v1beta1_keywords.py b/scripts/fixup_container_v1beta1_keywords.py index 561433ea..34053f77 100644 --- a/scripts/fixup_container_v1beta1_keywords.py +++ b/scripts/fixup_container_v1beta1_keywords.py @@ -72,7 +72,7 @@ class containerCallTransformer(cst.CSTTransformer): 'start_ip_rotation': ('project_id', 'zone', 'cluster_id', 'name', 'rotate_credentials', ), 'update_cluster': ('project_id', 'zone', 'cluster_id', 'update', 'name', ), 'update_master': ('project_id', 'zone', 'cluster_id', 'master_version', 'name', ), - 'update_node_pool': ('project_id', 'zone', 'cluster_id', 'node_pool_id', 'node_version', 'image_type', 'locations', 'workload_metadata_config', 'name', 'upgrade_settings', 'tags', 'taints', 'labels', 'linux_node_config', 'kubelet_config', 'node_network_config', 'gcfs_config', 'confidential_nodes', 'gvnic', 'logging_config', 'resource_labels', ), + 'update_node_pool': ('project_id', 'zone', 'cluster_id', 'node_pool_id', 'node_version', 'image_type', 'locations', 'workload_metadata_config', 'name', 'upgrade_settings', 'tags', 'taints', 'labels', 'linux_node_config', 'kubelet_config', 'node_network_config', 'gcfs_config', 'confidential_nodes', 'gvnic', 'fast_socket', 'logging_config', 'resource_labels', ), } def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: