Skip to content

Commit 9e6f299

Browse files
authored
deploy with snapshot plugin enabled (#203)
1 parent d53d105 commit 9e6f299

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

cmd/snapshot-tool/main.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@ import (
2121
kaischedulerfake "github.com/NVIDIA/KAI-scheduler/pkg/apis/client/clientset/versioned/fake"
2222
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/actions"
2323
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/cache"
24-
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/conf"
2524
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/conf_util"
2625
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/framework"
2726
"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/log"
@@ -82,7 +81,7 @@ func main() {
8281
schedulerCache.WaitForCacheSync(stopCh)
8382

8483
ssn, err := framework.OpenSession(
85-
schedulerCache, &conf.SchedulerConfiguration{}, snapshot.SchedulerParams, "", &http.ServeMux{},
84+
schedulerCache, snapshot.Config, snapshot.SchedulerParams, "", &http.ServeMux{},
8685
)
8786
if err != nil {
8887
log.InfraLogger.Fatalf(err.Error(), err)

deployments/kai-scheduler/templates/services/scheduler-configmap.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ data:
2121
- name: dynamicresources
2222
- name: gpupack
2323
- name: gpusharingorder
24+
- name: snapshot
2425
- name: nodeplacement
2526
arguments:
2627
cpu: {{ .Values.scheduler.placementStrategy }}
@@ -29,4 +30,4 @@ kind: ConfigMap
2930
metadata:
3031
labels:
3132
app: scheduler
32-
name: scheduler-config
33+
name: scheduler-config

0 commit comments

Comments
 (0)