fix memory scheduler

hephbuild · Apr 8, 2024 · 8e45bcf · 8e45bcf
1 parent 3cb8b1b
commit 8e45bcf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bootstrap/bootstrap.go b/bootstrap/bootstrap.go
@@ -143,8 +143,8 @@ type Bootstrap struct {
 
 func DefaultScheduler(cpu int) *worker2.ResourceScheduler {
 	return worker2.NewResourceScheduler(map[string]float64{
-		"cpu": float64(cpu),
-		"mem": float64(memory.TotalMemory()),
+		"cpu":    float64(cpu),
+		"memory": float64(memory.TotalMemory()),
 	}, map[string]float64{
 		"cpu": float64(1),
 	})

diff --git a/hbuiltin/target_spec_parser.go b/hbuiltin/target_spec_parser.go
@@ -196,8 +196,8 @@ func specFromArgs(args TargetArgs, pkg *packages.Package) (specs.Target, error)
 	}
 
 	t.Requests = map[string]float64{
-		"cpu": 1,
-		"mem": 100000000, // 1M
+		"cpu":    1,
+		"memory": 100000000, // 1M
 	}
 	for _, item := range args.Requests.Items() {
 		v, err := requestFromArg(item.Value)