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

updated octopus cardname 

Former-commit-id: a62d3350bd21690be3f75a11bfed6518fc23b0a8
This commit is contained in:
tzwang 2024-05-12 22:50:48 +08:00
parent f8a94bb090
commit 99985ec130
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -57,6 +57,9 @@ const (
TRAIN_CMD = "cd /code; python train.py" TRAIN_CMD = "cd /code; python train.py"
VERSION = "V1" VERSION = "V1"
DOMAIN = "http://192.168.242.41:8001/" DOMAIN = "http://192.168.242.41:8001/"
CAMBRICON_CN = "寒武纪290"
ENFLAME_CN = "燧原T20"
ILUVATAR_CN = "天数BI-V100"
) )
var ( var (
@ -65,6 +68,11 @@ var (
GCU: ENFLAME, GCU: ENFLAME,
BIV100: ILUVATAR, BIV100: ILUVATAR,
} }
cardCnMap = map[string]string{
MLU: CAMBRICON_CN,
GCU: ENFLAME_CN,
BIV100: ILUVATAR_CN,
}
cardTopsMap = map[string]float64{ cardTopsMap = map[string]float64{
MLU: CAMBRICONMLU290, MLU: CAMBRICONMLU290,
GCU: EnflameT20, GCU: EnflameT20,
@ -742,7 +750,7 @@ func setResourceIdByCard(option *option.AiOption, specs *octopus.GetResourceSpec
if spec.Price == 1 { if spec.Price == 1 {
ns := strings.Split(spec.Name, COMMA) ns := strings.Split(spec.Name, COMMA)
cardSpecs := strings.Split(ns[0], STAR) cardSpecs := strings.Split(ns[0], STAR)
if cardSpecs[1] == computeCard { if cardSpecs[1] == cardCnMap[computeCard] {
option.ResourceId = spec.Id option.ResourceId = spec.Id
option.ComputeCard = computeCard option.ComputeCard = computeCard
return nil return nil
@ -758,7 +766,7 @@ func setResourceIdByCard(option *option.AiOption, specs *octopus.GetResourceSpec
if spec.Price == 1 { if spec.Price == 1 {
ns := strings.Split(spec.Name, COMMA) ns := strings.Split(spec.Name, COMMA)
cardSpecs := strings.Split(ns[0], STAR) cardSpecs := strings.Split(ns[0], STAR)
if cardSpecs[1] == computeCard { if cardSpecs[1] == cardCnMap[computeCard] {
option.ResourceId = spec.Id option.ResourceId = spec.Id
option.ComputeCard = computeCard option.ComputeCard = computeCard
return nil return nil
@ -772,7 +780,7 @@ func setResourceIdByCard(option *option.AiOption, specs *octopus.GetResourceSpec
continue continue
} }
cardSpecs := strings.Split(ns[0], STAR) cardSpecs := strings.Split(ns[0], STAR)
if cardSpecs[1] != computeCard { if cardSpecs[1] != cardCnMap[computeCard] {
continue continue
} }
s, err := strconv.ParseFloat(cardSpecs[0], 64) s, err := strconv.ParseFloat(cardSpecs[0], 64)