resources

Documentation

Overview

Copyright (c) Microsoft Corporation. Licensed under the MIT license.

Index

• Constants
• func CheckNvidiaPlugin(ctx context.Context, nodeObj *corev1.Node) bool
• func CheckResourceStatus(obj client.Object, kubeClient client.Client, timeoutDuration time.Duration) error
• func CreateResource(ctx context.Context, resource client.Object, kubeClient client.Client) error
• func GenerateDeploymentManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, imageName string, ...) *appsv1.Deployment
• func GenerateDeploymentManifestWithPodTemplate(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, ...) *appsv1.Deployment
• func GenerateHeadlessServiceManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace) *corev1.Service
• func GenerateServiceManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, ...) *corev1.Service
• func GenerateStatefulSetManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, imageName string, ...) *appsv1.StatefulSet
• func GenerateTuningJobManifest(ctx context.Context, wObj *kaitov1alpha1.Workspace, imageName string, ...) *batchv1.Job
• func GetNode(ctx context.Context, nodeName string, kubeClient client.Client) (*corev1.Node, error)
• func GetResource(ctx context.Context, name, namespace string, kubeClient client.Client, ...) error
• func ListNodes(ctx context.Context, kubeClient client.Client, ...) (*corev1.NodeList, error)
• func UpdateNodeWithLabel(ctx context.Context, nodeName, labelKey, labelValue string, ...) error

Constants

const (
	LabelKeyNvidia    = "accelerator"
	LabelValueNvidia  = "nvidia"
	CapacityNvidiaGPU = "nvidia.com/gpu"
)

Functions

func CheckNvidiaPlugin

func CheckNvidiaPlugin(ctx context.Context, nodeObj *corev1.Node) bool

func CheckResourceStatus

func CheckResourceStatus(obj client.Object, kubeClient client.Client, timeoutDuration time.Duration) error

func CreateResource

func CreateResource(ctx context.Context, resource client.Object, kubeClient client.Client) error

func GenerateDeploymentManifest

func GenerateDeploymentManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, imageName string,
	imagePullSecretRefs []corev1.LocalObjectReference, replicas int, commands []string, containerPorts []corev1.ContainerPort,
	livenessProbe, readinessProbe *corev1.Probe, resourceRequirements corev1.ResourceRequirements,
	tolerations []corev1.Toleration, volumes []corev1.Volume, volumeMount []corev1.VolumeMount) *appsv1.Deployment

func GenerateDeploymentManifestWithPodTemplate

func GenerateDeploymentManifestWithPodTemplate(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, tolerations []corev1.Toleration) *appsv1.Deployment

func GenerateHeadlessServiceManifest

func GenerateHeadlessServiceManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace) *corev1.Service

func GenerateServiceManifest

func GenerateServiceManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, serviceType corev1.ServiceType, isStatefulSet bool) *corev1.Service

func GenerateStatefulSetManifest

func GenerateStatefulSetManifest(ctx context.Context, workspaceObj *kaitov1alpha1.Workspace, imageName string,
	imagePullSecretRefs []corev1.LocalObjectReference, replicas int, commands []string, containerPorts []corev1.ContainerPort,
	livenessProbe, readinessProbe *corev1.Probe, resourceRequirements corev1.ResourceRequirements,
	tolerations []corev1.Toleration, volumes []corev1.Volume, volumeMount []corev1.VolumeMount) *appsv1.StatefulSet

func GenerateTuningJobManifest

func GenerateTuningJobManifest(ctx context.Context, wObj *kaitov1alpha1.Workspace, imageName string,
	imagePullSecretRefs []corev1.LocalObjectReference, replicas int, commands []string, containerPorts []corev1.ContainerPort,
	livenessProbe, readinessProbe *corev1.Probe, resourceRequirements corev1.ResourceRequirements, tolerations []corev1.Toleration,
	initContainers []corev1.Container, sidecarContainers []corev1.Container, volumes []corev1.Volume, volumeMounts []corev1.VolumeMount,
	envVars []corev1.EnvVar) *batchv1.Job

func GetNode

func GetNode(ctx context.Context, nodeName string, kubeClient client.Client) (*corev1.Node, error)

GetNode get kubernetes node object with a provided name

func GetResource

func GetResource(ctx context.Context, name, namespace string, kubeClient client.Client, resource client.Object) error

func ListNodes

func ListNodes(ctx context.Context, kubeClient client.Client, labelSelector client.MatchingLabels) (*corev1.NodeList, error)

ListNodes get list of kubernetes nodes

func UpdateNodeWithLabel

func UpdateNodeWithLabel(ctx context.Context, nodeName, labelKey, labelValue string, kubeClient client.Client) error

UpdateNodeWithLabel update the node object with the label key/value