From a525685907b12c9bbfeadbfd8389dbfba3be46d1 Mon Sep 17 00:00:00 2001 From: tzwang Date: Fri, 13 Sep 2024 15:42:11 +0800 Subject: [PATCH] updated shuguang InferDeployInstance logic Former-commit-id: 8a1f7bd76a21d72b85e3ed59cb01579f9ae601ce --- .../service/inference/imageInference/imageInference.go | 3 +-- internal/storeLink/shuguangai.go | 3 +++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/internal/scheduler/service/inference/imageInference/imageInference.go b/internal/scheduler/service/inference/imageInference/imageInference.go index 700eb864..c2961812 100644 --- a/internal/scheduler/service/inference/imageInference/imageInference.go +++ b/internal/scheduler/service/inference/imageInference/imageInference.go @@ -240,7 +240,7 @@ func (i *ImageInference) inferImages(cs []*FilteredCluster) ([]*types.ImageResul var wg sync.WaitGroup var ch = make(chan *types.ImageResult, len(i.files)) var results []*types.ImageResult - limit := make(chan bool, 7) + limit := make(chan bool, 5) var imageNumIdx int32 = 0 var imageNumIdxEnd int32 = 0 @@ -396,7 +396,6 @@ func (i *ImageInference) sendInferReq(images []*ImageFile, cluster *FilteredClus return } }(image, cluster) - <-limit } } diff --git a/internal/storeLink/shuguangai.go b/internal/storeLink/shuguangai.go index 60d82880..809441d8 100644 --- a/internal/storeLink/shuguangai.go +++ b/internal/storeLink/shuguangai.go @@ -801,8 +801,11 @@ func (s *ShuguangAi) GetInferDeployInstanceList(ctx context.Context) ([]*inferen var insList []*inference.DeployInstance params := &hpcAC.GetInstanceServiceListReqParam{ InstanceServiceName: DEPLOY_INSTANCE_PREFIEX, + Status: "", + TaskType: "", Start: 0, Limit: DEPLOY_INSTANCE_LIMIT, + Sort: "desc", } req := &hpcacclient.GetInstanceServiceListReq{ Param: params,