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

fix: Complete the modelarts GetResourceStats of ai 

Former-commit-id: b23d85fa84fa5ec5583f4abc304ec65c72560e83
This commit is contained in:
qiwang 2024-07-29 16:17:24 +08:00
parent bb2b8263a8
commit 4ed1e829bb
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
internal/storeLink/modelarts.go
View File

@ -201,10 +201,11 @@ func (m *ModelArtsLink) GetResourceStats(ctx context.Context) (*collector.Resour
num32, _ := strconv.Atoi(resp1.Items[0].Spec.Npu.Size) num32, _ := strconv.Atoi(resp1.Items[0].Spec.Npu.Size)
var cards []*collector.Card var cards []*collector.Card
card := &collector.Card{ card := &collector.Card{
Platform: MODELARTS, Platform: MODELARTS,
Type: CARD, Type: CARD,
Name: Npu, Name: Npu,
CardNum: int32(num32), CardNum: int32(num32),
TOpsAtFp16: float64(num32 * 320),
} }
cards = append(cards, card) cards = append(cards, card)
resourceStats.CardsAvail = cards resourceStats.CardsAvail = cards

1
pkg/constants/task.go
View File

@ -28,4 +28,5 @@ const (
WaitStart = "WaitStart" WaitStart = "WaitStart"
Pending = "Pending" Pending = "Pending"
Stopped = "Stopped" Stopped = "Stopped"
Deploying = "Deploying"
) )