utils

package

v0.7.2 Latest Latest Go to latest Published: Jul 7, 2025 License: Apache-2.0 Imports: 9 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/NVIDIA/KAI-scheduler

Links

Documentation ¶

Index ¶

func GetAllPendingJobs(ssn *framework.Session) map[common_info.PodGroupID]*podgroup_info.PodGroupInfo
func GetMessageOfEviction(ssn *framework.Session, actionType framework.ActionType, ...) string
func IsEnoughGPUsAllocatableForJob(job *podgroup_info.PodGroupInfo, ssn *framework.Session, isRealAllocation bool) bool
type JobsOrderByQueues
- func GetVictimsQueue(ssn *framework.Session, filter func(*podgroup_info.PodGroupInfo) bool) *JobsOrderByQueues
- func NewJobsOrderByQueues(ssn *framework.Session, options JobsOrderInitOptions) JobsOrderByQueues
type JobsOrderInitOptions

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func GetAllPendingJobs ¶

func GetAllPendingJobs(ssn *framework.Session) map[common_info.PodGroupID]*podgroup_info.PodGroupInfo

func GetMessageOfEviction ¶

func GetMessageOfEviction(ssn *framework.Session, actionType framework.ActionType, preempteeTask *pod_info.PodInfo,
	preemptorJob *podgroup_info.PodGroupInfo) string

func IsEnoughGPUsAllocatableForJob ¶

func IsEnoughGPUsAllocatableForJob(job *podgroup_info.PodGroupInfo, ssn *framework.Session, isRealAllocation bool) bool

Types ¶

type JobsOrderByQueues ¶

type JobsOrderByQueues struct {
	// contains filtered or unexported fields
}

func GetVictimsQueue ¶

func GetVictimsQueue(
	ssn *framework.Session,
	filter func(*podgroup_info.PodGroupInfo) bool) *JobsOrderByQueues

func NewJobsOrderByQueues ¶

func NewJobsOrderByQueues(ssn *framework.Session, options JobsOrderInitOptions) JobsOrderByQueues

func (*JobsOrderByQueues) InitializeWithJobs ¶

func (jobsOrder *JobsOrderByQueues) InitializeWithJobs(
	jobsToOrder map[common_info.PodGroupID]*podgroup_info.PodGroupInfo)

func (*JobsOrderByQueues) IsEmpty ¶

func (jobsOrder *JobsOrderByQueues) IsEmpty() bool

func (*JobsOrderByQueues) Len ¶

func (jobsOrder *JobsOrderByQueues) Len() int

func (*JobsOrderByQueues) PopNextJob ¶

func (jobsOrder *JobsOrderByQueues) PopNextJob() *podgroup_info.PodGroupInfo

func (*JobsOrderByQueues) PushJob ¶

func (jobsOrder *JobsOrderByQueues) PushJob(job *podgroup_info.PodGroupInfo)

type JobsOrderInitOptions ¶

type JobsOrderInitOptions struct {
	FilterUnready            bool
	FilterNonPending         bool
	FilterNonPreemptible     bool
	FilterNonActiveAllocated bool
	VictimQueue              bool
	MaxJobsQueueDepth        int
}

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL