Add DRA and gpu partition mode for GPU (#1215)

Author: thearyanahmed

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: