Skip to content

M
mogo
Commit e89eab66 authored by vicotor's avatar vicotor
Browse files
Options

update mogo

parent 694be5e5
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
operator/workerInstalledInfo.go
View file @ e89eab66
...@@ -95,7 +95,7 @@ func (d *WorkerInstalledOperator) FindWorkerByModelIdAndGpuMem(ctx context.Conte ...@@ -95,7 +95,7 @@ func (d *WorkerInstalledOperator) FindWorkerByModelIdAndGpuMem(ctx context.Conte
// sort by gpu_free // sort by gpu_free
findOptions := options.Find() findOptions := options.Find()
findOptions.SetLimit(int64(limit)) findOptions.SetLimit(int64(limit))
findOptions.SetSort(bson.D{{"gpu_free", 1}}) findOptions.SetSort(bson.D{{"gpu_free", -1}})
selector := bson.M{"model_id": modelId, "gpu_free": bson.M{"$gte": mem}} selector := bson.M{"model_id": modelId, "gpu_free": bson.M{"$gte": mem}}
cursor, err := d.col.Find(ctx, selector, findOptions) cursor, err := d.col.Find(ctx, selector, findOptions)
......
operator/workerinfo.go
View file @ e89eab66
...@@ -201,7 +201,7 @@ func (d *WorkerInfoOperator) FindWorkerByRunningModelAndSortByWaitTime(ctx conte ...@@ -201,7 +201,7 @@ func (d *WorkerInfoOperator) FindWorkerByRunningModelAndSortByWaitTime(ctx conte
// sort by wait time // sort by wait time
findOptions := options.Find() findOptions := options.Find()
findOptions.SetLimit(int64(limit)) findOptions.SetLimit(int64(limit))
findOptions.SetSort(bson.D{{"model_infos.running_models.wait_time", 1}}) findOptions.SetSort(bson.D{{"model_infos.running_models.wait_time", -1}})
selector := bson.M{"model_infos.running_models.model_id": modelId} selector := bson.M{"model_infos.running_models.model_id": modelId}
cursor, err := d.col.Find(ctx, selector, findOptions) cursor, err := d.col.Find(ctx, selector, findOptions)
...@@ -222,7 +222,7 @@ func (d *WorkerInfoOperator) FindWorkerByInstallModelAndSortByGpuRam(ctx context ...@@ -222,7 +222,7 @@ func (d *WorkerInfoOperator) FindWorkerByInstallModelAndSortByGpuRam(ctx context
// sort by gpu ram // sort by gpu ram
findOptions := options.Find() findOptions := options.Find()
findOptions.SetLimit(int64(limit)) findOptions.SetLimit(int64(limit))
findOptions.SetSort(bson.D{{"hardware.gpu.ram", 1}}) findOptions.SetSort(bson.D{{"hardware.gpu.ram", -1}})
selector := bson.M{"model_infos.installed_models.model_id": modelId, "hardware.gpu.performance": bson.M{"$gte": performance}, "hardware.gpu.ram": bson.M{"$gte": ram}} selector := bson.M{"model_infos.installed_models.model_id": modelId, "hardware.gpu.performance": bson.M{"$gte": performance}, "hardware.gpu.ram": bson.M{"$gte": ram}}
cursor, err := d.col.Find(ctx, selector, findOptions) cursor, err := d.col.Find(ctx, selector, findOptions)
......
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