Browse Source

fix:Solve the problem of null pointer of modelarts

pull/465/head
qiwang 7 months ago
parent
commit
331249a223
1 changed files with 14 additions and 8 deletions
  1. +14
    -8
      internal/storeLink/modelarts.go

+ 14
- 8
internal/storeLink/modelarts.go View File

@@ -896,8 +896,12 @@ func (m *ModelArtsLink) GetInferDeployInstance(ctx context.Context, id string) (
ins.ClusterName = m.platform
ins.CreatedTime = string(resp.StartTime)
ins.ClusterType = TYPE_MODELARTS
ins.ModelName = resp.Config[0].ModelName
ins.ModelType = m.ModelType
if resp.Config != nil {
ins.ModelName = resp.Config[0].ModelName
}
if m.ModelType != "" {
ins.ModelType = m.ModelType
}
ins.InferUrl = resp.AccessAddress
return ins, nil
}
@@ -1071,12 +1075,14 @@ func (m *ModelArtsLink) GetResourceSpecs(ctx context.Context) (*collector.Resour
wg.Done()
}
if TrainLists.Status.Phase == "Running" {
CoreNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Cpu.CoreNum
NpuNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Npu.UnitNum
MemoryNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Memory.Size
VMemoryNum, _ = strconv.ParseInt(TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Npu.Memory, 10, 64)
VMemoryNum += VMemoryNum
RunningTaskNum += 1
if TrainLists.Spec.Resource.FlavorDetail != nil {
CoreNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Cpu.CoreNum
NpuNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Npu.UnitNum
MemoryNum += TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Memory.Size
VMemoryNum, _ = strconv.ParseInt(TrainLists.Spec.Resource.FlavorDetail.FlavorInfo.Npu.Memory, 10, 64)
VMemoryNum += VMemoryNum
RunningTaskNum += 1
}
}
}
MoUsage.CpuAvailable = cpusum - int64(CoreNum)


Loading…
Cancel
Save