Add DRA and gpu partition mode for GPU

specification/resources/kubernetes/models/amd_gpu_dra_driver.yml

@@ -0,0 +1,8 @@
+type: object
+nullable: true
+description: An object specifying whether the AMD GPU DRA Driver should be enabled in the Kubernetes cluster. Mutually exclusive with `amd_gpu_device_plugin`.
+properties:
+  enabled:
+    type: boolean
+    description: Indicates whether the AMD GPU DRA Driver is enabled.
+    example: true

specification/resources/kubernetes/models/cluster.yml

@@ -203,6 +203,12 @@ properties:
   nvidia_gpu_device_plugin:
     $ref: "nvidia_gpu_device_plugin.yml"
 
+  nvidia_gpu_dra_driver:
+    $ref: "nvidia_gpu_dra_driver.yml"
+
+  amd_gpu_dra_driver:
+    $ref: "amd_gpu_dra_driver.yml"
+
   rdma_shared_dev_plugin:
     $ref: "rdma_shared_dev_plugin.yml"
 

specification/resources/kubernetes/models/cluster_read.yml

@@ -207,6 +207,12 @@ properties:
   nvidia_gpu_device_plugin:
     $ref: "nvidia_gpu_device_plugin.yml"
 
+  nvidia_gpu_dra_driver:
+    $ref: "nvidia_gpu_dra_driver.yml"
+
+  amd_gpu_dra_driver:
+    $ref: "amd_gpu_dra_driver.yml"
+
   rdma_shared_dev_plugin:
     $ref: "rdma_shared_dev_plugin.yml"
 

specification/resources/kubernetes/models/cluster_update.yml

@@ -71,6 +71,12 @@ properties:
   nvidia_gpu_device_plugin:
     $ref: "nvidia_gpu_device_plugin.yml"
 
+  nvidia_gpu_dra_driver:
+    $ref: "nvidia_gpu_dra_driver.yml"
+
+  amd_gpu_dra_driver:
+    $ref: "amd_gpu_dra_driver.yml"
+
   rdma_shared_dev_plugin:
     $ref: "rdma_shared_dev_plugin.yml"
 

specification/resources/kubernetes/models/node_pool.yml

@@ -3,6 +3,7 @@ kubernetes_node_pool:
   allOf:
   - $ref: "#/kubernetes_node_pool_size"
   - $ref: "#/kubernetes_node_pool_base"
+  - $ref: "#/kubernetes_node_pool_gpu_partition"
 
   required:
   - name
@@ -27,6 +28,20 @@ kubernetes_node_pool_size:
       description: The slug identifier for the type of Droplet used as workers in
         the node pool.
 
+kubernetes_node_pool_gpu_partition:
+  type: object
+  properties:
+    gpu_partition_mode:
+      type: string
+      enum:
+      - AMD_PARTITION_MODE_SPX_NPS1
+      - AMD_PARTITION_MODE_DPX_NPS2
+      example: AMD_PARTITION_MODE_SPX_NPS1
+      description: The AMD GPU partition mode for this node pool. Only applicable
+        to AMD GPU sizes that support partitioning. Immutable after the node
+        pool is created. When omitted, the GPUs in the pool are left
+        unpartitioned.
+
 kubernetes_node_pool_base:
   type: object
   properties:

specification/resources/kubernetes/models/nvidia_gpu_dra_driver.yml

@@ -0,0 +1,8 @@
+type: object
+nullable: true
+description: An object specifying whether the NVIDIA GPU DRA Driver should be enabled in the Kubernetes cluster. Mutually exclusive with `nvidia_gpu_device_plugin`.
+properties:
+  enabled:
+    type: boolean
+    description: Indicates whether the NVIDIA GPU DRA Driver is enabled.
+    example: true

specification/resources/kubernetes/responses/examples.yml

@@ -126,6 +126,10 @@ kubernetes_clusters_all:
         enabled: false
       nvidia_gpu_device_plugin:
         enabled: false
+      nvidia_gpu_dra_driver:
+        enabled: false
+      amd_gpu_dra_driver:
+        enabled: false
       rdma_shared_dev_plugin:
         enabled: false
       coredns_autoscaler:
@@ -263,6 +267,10 @@ kubernetes_single:
         enabled: false
       nvidia_gpu_device_plugin:
         enabled: false
+      nvidia_gpu_dra_driver:
+        enabled: false
+      amd_gpu_dra_driver:
+        enabled: false
       rdma_shared_dev_plugin:
         enabled: false
       coredns_autoscaler:
@@ -397,6 +405,10 @@ kubernetes_updated:
         enabled: false
       nvidia_gpu_device_plugin:
         enabled: false
+      nvidia_gpu_dra_driver:
+        enabled: false
+      amd_gpu_dra_driver:
+        enabled: false
       rdma_shared_dev_plugin:
         enabled: false
       coredns_autoscaler:
@@ -496,6 +508,10 @@ kubernetes_clusters_create_basic_response:
         enabled: false
       nvidia_gpu_device_plugin:
         enabled: false
+      nvidia_gpu_dra_driver:
+        enabled: false
+      amd_gpu_dra_driver:
+        enabled: false
       rdma_shared_dev_plugin:
         enabled: false
       coredns_autoscaler: