|
|
|
@@ -29,6 +29,7 @@ import ( |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/utils/jcs" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/strategy" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/strategy/param" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/storeLink" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/constants" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/response" |
|
|
|
@@ -37,6 +38,7 @@ import ( |
|
|
|
"gitlink.org.cn/JointCloud/pcm-octopus/octopus" |
|
|
|
"gitlink.org.cn/JointCloud/pcm-openi/model" |
|
|
|
"strconv" |
|
|
|
"strings" |
|
|
|
"sync" |
|
|
|
) |
|
|
|
|
|
|
|
@@ -343,7 +345,14 @@ func updateAiOptionByMode(cluster *strategy.AssignedCluster, opt *option.AiOptio |
|
|
|
|
|
|
|
opt.ResourcesRequired = cluster.ResourcesRequired |
|
|
|
|
|
|
|
opt.Output = cluster.Output |
|
|
|
// assume params include output for modelarts |
|
|
|
for _, param := range cluster.Params { |
|
|
|
s := strings.Split(param, storeLink.COMMA) |
|
|
|
if s[0] == "output" { |
|
|
|
opt.Output = s[1] |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
default: |
|
|
|
|
|
|
|
} |
|
|
|
|