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

modified octopus functions generateAlgorithmId 

Former-commit-id: 0440af0e370819df4d58399d89574ac29c62c44f
This commit is contained in:
tzwang 2024-03-07 17:28:43 +08:00
parent 32bee79af3
commit 6f181c313a
1 changed files with 30 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
api/internal/storeLink/octopus.go
View File

@ -341,11 +341,10 @@ func (o *OctopusLink) generateResourceId(option *option.AiOption) error {
return nil return nil
} }
} }
} }
if option.ResourceType == CARD { if option.ResourceType == CARD {
err = setResourceIdByCard(option, specResp, MLU) err = setResourceIdByCard(option, specResp, GCU)
if err != nil { if err != nil {
return err return err
} }
@ -418,16 +417,23 @@ func (o *OctopusLink) generateImageId(option *option.AiOption) error {
if !preImgResp.Success { if !preImgResp.Success {
return errors.New("failed to get PresetImages") return errors.New("failed to get PresetImages")
} }
for _, image := range preImgResp.Payload.Images {
if strings.Contains(image.ImageName, option.TaskType) && strings.Contains(image.ImageName, cardAliasMap[option.ComputeCard]) { if option.ResourceType == CARD {
option.ImageId = image.Id for _, image := range preImgResp.Payload.Images {
return nil if strings.Contains(image.ImageName, cardAliasMap[option.ComputeCard]) {
option.ImageId = image.Id
return nil
}
} }
} }
return errors.New("failed to get ImageId") return errors.New("failed to get ImageId")
} }
func (o *OctopusLink) generateAlgorithmId(option *option.AiOption) error { func (o *OctopusLink) generateAlgorithmId(option *option.AiOption) error {
// temporarily set algorithm to cnn
option.AlgorithmName = "cnn"
req := &octopus.GetMyAlgorithmListReq{ req := &octopus.GetMyAlgorithmListReq{
Platform: o.platform, Platform: o.platform,
PageIndex: o.pageIndex, PageIndex: o.pageIndex,
@ -442,18 +448,33 @@ func (o *OctopusLink) generateAlgorithmId(option *option.AiOption) error {
} }
for _, algorithm := range resp.Payload.Algorithms { for _, algorithm := range resp.Payload.Algorithms {
if algorithm.FrameworkName == strings.Title(option.TaskType) && strings.Contains(algorithm.AlgorithmName, option.DatasetsName) { if algorithm.FrameworkName == strings.Title(option.TaskType) {
ns := strings.Split(algorithm.AlgorithmName, DASH)
if ns[0] != option.DatasetsName {
continue
}
if ns[1] != option.AlgorithmName {
continue
}
if ns[2] != option.ResourceType {
continue
}
option.AlgorithmId = algorithm.AlgorithmId option.AlgorithmId = algorithm.AlgorithmId
return nil return nil
} }
} }
return nil return errors.New("failed to get AlgorithmId")
} }
func (o *OctopusLink) generateCmd(option *option.AiOption) error { func (o *OctopusLink) generateCmd(option *option.AiOption) error {
if option.Cmd == "" { if option.Cmd == "" {
option.Cmd = TRAIN_CMD switch option.ComputeCard {
case GCU:
option.Cmd = "cd /code; python3 train.py"
default:
option.Cmd = TRAIN_CMD
}
} }
return nil return nil