workloadutils

package

v0.1.2 Latest Latest Go to latest Published: Apr 8, 2025 License: Apache-2.0 Imports: 16 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/silogen/kaiwo

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
Variables
func AddEntrypoint(entrypoint string, podTemplateSpec *corev1.PodTemplateSpec) error
func CheckPodStatus(ctx context.Context, k8sClient client.Client, name string, namespace string, ...) (earliestRunningTime *metav1.Time, status v1alpha1.Status, err error)
func FillPodResources(podSpec *corev1.PodSpec, resources *corev1.ResourceRequirements, override bool)
func GetEarliestPodStartTime(ctx context.Context, k8sClient client.Client, name string, namespace string) *metav1.Time
func GetPodTemplate(dshmSize resource.Quantity, dangerous bool, ...) corev1.PodTemplateSpec
func GetRayClusterTemplate(dangerous bool, resourceRequirements v1.ResourceRequirements) *rayv1.RayClusterSpec
func UpdatePodSpec(kaiwoCommonMetaSpec v1alpha1.CommonMetaSpec, ...) error
func ValidateKaiwoResourceBeforeCreateOrUpdate(ctx context.Context, actual client.Object, kaiwoObjectMeta metav1.ObjectMeta) (*ctrl.Result, error)
type DownloadJobConfigMapReconciler
- func NewDownloadJobConfigMapReconciler(objectKey client.ObjectKey, storageSpec *v1alpha1.StorageSpec) *DownloadJobConfigMapReconciler
- func (r *DownloadJobConfigMapReconciler) Build(ctx context.Context, _ client.Client) (*corev1.ConfigMap, error)
- func (r *DownloadJobConfigMapReconciler) GetEmptyObject() *corev1.ConfigMap
type DownloadJobReconciler
- func NewDownloadJobReconciler(objectKey client.ObjectKey, storageSpec *v1alpha1.StorageSpec, ...) *DownloadJobReconciler

Constants ¶

View Source

const (
	KaiwoDownloadTypeLabelValue = "downloader"
)

Variables ¶

View Source

var (
	DefaultDataMountPath = baseutils.GetEnv("DEFAULT_DATA_MOUNT_PATH", "/workload")
	DefaultHfMountPath   = baseutils.GetEnv("DEFAULT_HF_MOUNT_PATH", "/hf_cache")
)

View Source

var (
	DefaultMemory = resource.MustParse("16Gi")
	DefaultCPU    = resource.MustParse("2")
)

View Source

var DefaultGpuResourceKey = baseutils.GetEnv("DEFAULT_GPU_RESOURCE_KEY", "amd.com/gpu")

Functions ¶

func AddEntrypoint ¶

func AddEntrypoint(entrypoint string, podTemplateSpec *corev1.PodTemplateSpec) error

AddEntrypoint updates the entrypoint command in the PodTemplateSpec.

func CheckPodStatus ¶

func CheckPodStatus(ctx context.Context, k8sClient client.Client, name string, namespace string, startTime *metav1.Time) (earliestRunningTime *metav1.Time, status v1alpha1.Status, err error)

func FillPodResources ¶

func FillPodResources(podSpec *corev1.PodSpec, resources *corev1.ResourceRequirements, override bool)

FillPodResources fills pod resources with a given template if they are not already set

func GetEarliestPodStartTime ¶

func GetEarliestPodStartTime(ctx context.Context, k8sClient client.Client, name string, namespace string) *metav1.Time

func GetPodTemplate ¶

func GetPodTemplate(dshmSize resource.Quantity, dangerous bool, resources corev1.ResourceRequirements, workloadContainerName string) corev1.PodTemplateSpec

func GetRayClusterTemplate ¶

func GetRayClusterTemplate(dangerous bool, resourceRequirements v1.ResourceRequirements) *rayv1.RayClusterSpec

func UpdatePodSpec ¶

func UpdatePodSpec(kaiwoCommonMetaSpec v1alpha1.CommonMetaSpec, labelContext common.KaiwoLabelContext, template *corev1.PodTemplateSpec, name string, replicas int, gpusPerReplica int, override bool) error

func ValidateKaiwoResourceBeforeCreateOrUpdate ¶

func ValidateKaiwoResourceBeforeCreateOrUpdate(ctx context.Context, actual client.Object, kaiwoObjectMeta metav1.ObjectMeta) (*ctrl.Result, error)

Types ¶

type DownloadJobConfigMapReconciler ¶

type DownloadJobConfigMapReconciler struct {
	common.ResourceReconcilerBase[*corev1.ConfigMap]
	StorageSpec *v1alpha1.StorageSpec
}

func NewDownloadJobConfigMapReconciler ¶

func NewDownloadJobConfigMapReconciler(objectKey client.ObjectKey, storageSpec *v1alpha1.StorageSpec) *DownloadJobConfigMapReconciler

func (*DownloadJobConfigMapReconciler) Build ¶

func (r *DownloadJobConfigMapReconciler) Build(ctx context.Context, _ client.Client) (*corev1.ConfigMap, error)

func (*DownloadJobConfigMapReconciler) GetEmptyObject ¶

func (r *DownloadJobConfigMapReconciler) GetEmptyObject() *corev1.ConfigMap

type DownloadJobReconciler ¶

type DownloadJobReconciler struct {
	common.ResourceReconcilerBase[*batchv1.Job]
	StorageSpec *v1alpha1.StorageSpec
	PvcBaseName string
	UserEnvVars []corev1.EnvVar
}

func NewDownloadJobReconciler ¶

func NewDownloadJobReconciler(objectKey client.ObjectKey, storageSpec *v1alpha1.StorageSpec, pvcBaseName string, userEnvVars []corev1.EnvVar) *DownloadJobReconciler

func (*DownloadJobReconciler) Build ¶

func (r *DownloadJobReconciler) Build(_ context.Context, _ client.Client) (*batchv1.Job, error)

func (*DownloadJobReconciler) GetEmptyObject ¶

func (r *DownloadJobReconciler) GetEmptyObject() *batchv1.Job

func (*DownloadJobReconciler) ShouldContinue ¶

func (r *DownloadJobReconciler) ShouldContinue(ctx context.Context, actual *batchv1.Job) *ctrl.Result

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL