Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add GPU Enabled flags
Browse files Browse the repository at this point in the history
ajanikow committed Apr 25, 2024
1 parent 5a9c96f commit 4be36ba
Showing 4 changed files with 11 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docs/api/ArangoMLExtension.V1Alpha1.md
Original file line number Diff line number Diff line change
@@ -100,7 +100,7 @@ Links:

Type: `boolean` <sup>[\[ref\]](https://github.com/arangodb/kube-arangodb/blob/1.2.40/pkg/apis/ml/v1alpha1/extension_spec_deployment.go#L52)</sup>

GPU defined if GPU Jobs are enabledt.
GPU defined if GPU Jobs are enabled.

Default Value: `false`

2 changes: 1 addition & 1 deletion pkg/apis/ml/v1alpha1/extension_spec_deployment.go
Original file line number Diff line number Diff line change
@@ -47,7 +47,7 @@ type ArangoMLExtensionSpecDeployment struct {
// Container Keeps the information about Container configuration
*schedulerContainerApi.Container `json:",inline"`

// GPU defined if GPU Jobs are enabledt.
// GPU defined if GPU Jobs are enabled.
// +doc/default: false
GPU *bool `json:"gpu,omitempty"`

8 changes: 8 additions & 0 deletions pkg/apis/ml/v1alpha1/extension_spec_job.go
Original file line number Diff line number Diff line change
@@ -36,6 +36,14 @@ const (
MLJobFeaturizationType JobType = "featurization"
)

func MLJobTypes() []JobType {
return []JobType{
MLJobTrainingType,
MLJobPredictionType,
MLJobFeaturizationType,
}
}

type ArangoMLJobsTemplates struct {
// Prediction defines template for the prediction job
Prediction *ArangoMLJobTemplates `json:"prediction,omitempty"`
2 changes: 1 addition & 1 deletion pkg/crd/crds/ml-extension.schema.generated.yaml
Original file line number Diff line number Diff line change
@@ -419,7 +419,7 @@ v1alpha1:
type: object
type: array
gpu:
description: GPU defined if GPU Jobs are enabledt.
description: GPU defined if GPU Jobs are enabled.
type: boolean
hostIPC:
type: boolean

0 comments on commit 4be36ba

Please sign in to comment.