jobparser for crd TrainingJob

.gitignore

@@ -1,3 +1,4 @@
 *~
 vendor/
 .glide/
+.vscode/

pkg/apis/paddlepaddle/v1/trainingjob.go

@@ -0,0 +1,32 @@
+package v1
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+// Elastic returns true if the job can scale to more workers.
+func (s *TrainingJob) Elastic() bool {
+	return s.Spec.Trainer.MinInstance < s.Spec.Trainer.MaxInstance
+}
+
+// GPU convert Resource Limit Quantity to int
+func (s *TrainingJob) GPU() int {
+	q := s.Spec.Trainer.Resources.Limits.NvidiaGPU()
+	gpu, ok := q.AsInt64()
+	if !ok {
+		// FIXME: treat errors
+		gpu = 0
+	}
+	return int(gpu)
+}
+
+// NeedGPU returns true if the job need GPU resource to run.
+func (s *TrainingJob) NeedGPU() bool {
+	return s.GPU() > 0
+}
+
+func (s *TrainingJob) String() string {
+	b, _ := json.MarshalIndent(s, "", "   ")
+	return fmt.Sprintf("%s", b)
+}

pkg/apis/paddlepaddle/v1/types.go

@@ -55,6 +55,7 @@ type TrainingJobSpec struct {
 	Passes            int                  `json:"passes,omitempty"`
 	Volumes           []corev1.Volume      `json:"volumes"`
 	VolumeMounts      []corev1.VolumeMount `json:"VolumeMounts"`
+	//TODO simplify the structure of sub-resource(mengyang)
 	//TrainingJob components.
 	Master  MasterSpec  `json:"master"`
 	Pserver PserverSpec `json:"pserver"`
@@ -111,12 +112,12 @@ type TrainerJobScaleStatus struct {
 type TrainingResourceType string
 
 const (
-	// MASTER is the master name of TrainingResourceType.
-	MASTER TrainingResourceType = "MASTER"
-	// PSERVER is the pserver name of TrainingResourceType.
-	PSERVER TrainingResourceType = "PSERVER"
-	// TRAINER is the trainer name of TrainingResourceType.
-	TRAINER TrainingResourceType = "TRAINER"
+	// Master is the master name of TrainingResourceType.
+	Master TrainingResourceType = "MASTER"
+	// Pserver is the pserver name of TrainingResourceType.
+	Pserver TrainingResourceType = "PSERVER"
+	// Trainer is the trainer name of TrainingResourceType.
+	Trainer TrainingResourceType = "TRAINER"
 )
 
 // ResourceState is the state of a type of resource