Skip to content

S
scheduler
Commit 0c26adba authored by vicotor's avatar vicotor
Browse files
Options

update

parent bb8df440
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
workerpoper/v2.go
View file @ 0c26adba
...@@ -82,8 +82,10 @@ func (p *poperV2) PopWorker(ctx context.Context, rdb *redis.Client, task *odysse ...@@ -82,8 +82,10 @@ func (p *poperV2) PopWorker(ctx context.Context, rdb *redis.Client, task *odysse
validWorkers = append(validWorkers, w) validWorkers = append(validWorkers, w)
} }
if len(validWorkers) == 0 { if len(validWorkers) == 0 {
// todo: set task.gpu_mem param.
var taskmem = int64(10)
// 2. select worker who has installed model. // 2. select worker who has installed model.
workers, err := p.workerInstalledOperator.FindWorkerByModelId(ctx, int(task.TaskType), 10) workers, err := p.workerInstalledOperator.FindWorkerByModelIdAndGpuMem(ctx, int(task.TaskType), taskmem, 10)
if err != nil { if err != nil {
log.WithField("tasktype", task.TaskType).WithError(err).Error("get installed model worker failed") log.WithField("tasktype", task.TaskType).WithError(err).Error("get installed model worker failed")
return types.Worker{}, err return types.Worker{}, err
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment