Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions gke/kcl.mod
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
[package]
name = "gke"
edition = "v0.11.3"
version = "0.0.3"
edition = "v0.11.4"
version = "0.0.4"

[dependencies]
k8s = "1.32.4"
39 changes: 39 additions & 0 deletions gke/v1/cloud_google_com_v1_compute_class.k
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,40 @@ schema CloudGoogleComV1ComputeClassSpecPrioritiesItems0:
minMemoryGb >= 0 if minMemoryGb not in [None, Undefined]


schema CloudGoogleComV1ComputeClassSpecPrioritiesItems0GpuSharing:
r"""
GpuSharing defines GPU sharing and partitioning config for a node.
Requires GKE version 1.35.2-gke.1485000 or later.

Attributes
----------
sharingStrategy : str, default is Undefined, optional
SharingStrategy describes the GPU sharing strategy.
Supported values: TIME_SHARING, MPS.
maxSharedClientsPerGPU : int, default is Undefined, optional
MaxSharedClientsPerGPU describes the maximum number of containers
that can share a GPU. Valid range: 2-48.
Required when sharingStrategy is set.
gpuPartitionSize : str, default is Undefined, optional
GpuPartitionSize enables multi-instance GPU (MIG) partitioning.
Sets the MIG profile applied to each GPU on the node.
Example values for nvidia-rtx-pro-6000: "1g.24gb", "2g.48gb", "4g.96gb".
Example values for nvidia-h100-80gb: "1g.10gb", "2g.20gb", "3g.40gb", "4g.40gb", "7g.80gb".
If omitted the GPU is not partitioned.
"""

sharingStrategy?: str

maxSharedClientsPerGPU?: int

gpuPartitionSize?: str


check:
maxSharedClientsPerGPU >= 2 if maxSharedClientsPerGPU not in [None, Undefined]
maxSharedClientsPerGPU <= 48 if maxSharedClientsPerGPU not in [None, Undefined]


schema CloudGoogleComV1ComputeClassSpecPrioritiesItems0Gpu:
r"""
Gpu defines preferred GPU config for a node.
Expand All @@ -257,6 +291,9 @@ schema CloudGoogleComV1ComputeClassSpecPrioritiesItems0Gpu:
DriverVersion describes version of GPU driver for a node.
$type : str, default is Undefined, optional
Type describes preferred GPU accelerator type for a node.
gpuSharing : CloudGoogleComV1ComputeClassSpecPrioritiesItems0GpuSharing, default is Undefined, optional
GpuSharing configures GPU sharing strategies and MIG partitioning.
Requires GKE version 1.35.2-gke.1485000 or later.
"""


Expand All @@ -266,6 +303,8 @@ schema CloudGoogleComV1ComputeClassSpecPrioritiesItems0Gpu:

$type?: str

gpuSharing?: CloudGoogleComV1ComputeClassSpecPrioritiesItems0GpuSharing


check:
count >= 1 if count not in [None, Undefined]
Expand Down
Loading