floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified shuguangAi implementations 

Former-commit-id: ec3fe090a5fc92df53e058aa6602923dca56e089
This commit is contained in:
tzwang 2024-02-02 11:28:19 +08:00
parent 41474ecaff
commit e2ca3b5be4
3 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/internal/scheduler/schedulers/option/aiOption.go
View File

@ -1,9 +1,11 @@
package option package option
type AiOption struct { type AiOption struct {
AiType string // shuguangAi/octopus //AiType string // shuguangAi/octopus
ResourceType string // cpu/gpu/compute card ResourceType string // cpu/gpu/compute card
TaskType string // pytorch/tensorflow TaskType string // pytorch/tensorflow
DatasetsType string
CodeType string
ImageId string ImageId string
SpecId string SpecId string

2
api/internal/scheduler/strategy/replication.go
View File

@ -1,7 +1,7 @@
package strategy package strategy
import ( import (
"github.com/pkg/errors" "errors"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/entity" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/entity"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/strategy/param" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/strategy/param"
) )

7
api/internal/storeLink/shuguangai.go
View File

@ -210,9 +210,14 @@ func (o *ShuguangAi) GetResourceSpecs() (*collector.ResourceSpecs, error) {
} }
func (o *ShuguangAi) Execute(option *option.AiOption) (interface{}, error) { func (o *ShuguangAi) Execute(option *option.AiOption) (interface{}, error) {
task, err := o.SubmitTask(option.ImageId, option.Cmd, option.Envs, option.Params, option.ResourceId, option.AiType) o.generateSubmitParams(option)
task, err := o.SubmitTask(option.ImageId, option.Cmd, option.Envs, option.Params, option.ResourceId, option.TaskType)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return task, nil return task, nil
} }
func (o *ShuguangAi) generateSubmitParams(option *option.AiOption) {
}