core

package

v0.0.0-...-12e0a72 Latest Latest Go to latest Published: Apr 21, 2024 License: Apache-2.0 Imports: 14 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/nebuly-ai/nos

Links

Open Source Insights

Documentation ¶

Index ¶

func IsNodeInitialized(node v1.Node) bool
func NewPartitioningPlanId() string
type Actuator
- func NewActuator(client client.Client, partitioner Partitioner) Actuator
type NodeInitializer
type PartitionCalculator
type PartitionableNode
type Partitioner
type PartitioningPlan
- func NewPartitioningPlan(s state.PartitioningState) PartitioningPlan
- func (p PartitioningPlan) GetId() string
type Planner
- func NewPlanner(partitioner PartitionCalculator, sliceCalculator gpu.SliceCalculator, ...) Planner
type SliceTracker
- func NewSliceTracker(snapshot Snapshot, calculator gpu.SliceCalculator, pods []v1.Pod) SliceTracker
- func (t SliceTracker) GetLackingSlices() map[gpu.Slice]int
- func (t SliceTracker) GetRequestedSlices() map[gpu.Slice]int
- func (t SliceTracker) Remove(pod v1.Pod)
type Snapshot
- func NewClusterSnapshot(nodes map[string]PartitionableNode, partitioner PartitionCalculator, ...) Snapshot
type SnapshotTaker
type Sorter
type SorterAdapter
- func NewPodSorter(sliceCalculator gpu.SliceCalculator) SorterAdapter
- func (f SorterAdapter) Sort(pods []v1.Pod) []v1.Pod

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func IsNodeInitialized ¶

func IsNodeInitialized(node v1.Node) bool

IsNodeInitialized checks if the GPU Partitioning on the provided node has already been initialized is initialized. A node is initialized if it has GPU Spec partitioning annotations, and according to these annotations all the GPUs of the node have at least one GPU partition.

func NewPartitioningPlanId ¶

func NewPartitioningPlanId() string

Types ¶

type Actuator ¶

type Actuator interface {
	Apply(ctx context.Context, snapshot Snapshot, plan PartitioningPlan) (bool, error)
}

func NewActuator ¶

func NewActuator(client client.Client, partitioner Partitioner) Actuator

type NodeInitializer ¶

type NodeInitializer interface {
	InitNodePartitioning(ctx context.Context, node v1.Node) error
}

type PartitionCalculator ¶

type PartitionCalculator interface {
	GetPartitioning(node PartitionableNode) state.NodePartitioning
}

type PartitionableNode ¶

type PartitionableNode interface {
	UpdateGeometryFor(slices map[gpu.Slice]int) (bool, error)
	GetName() string
	Geometry() map[gpu.Slice]int
	NodeInfo() framework.NodeInfo
	Clone() interface{}
	AddPod(pod v1.Pod) error
	HasFreeCapacity() bool
}

type Partitioner ¶

type Partitioner interface {
	ApplyPartitioning(ctx context.Context, node v1.Node, planId string, partitioning state.NodePartitioning) error
}

type PartitioningPlan ¶

type PartitioningPlan struct {
	DesiredState state.PartitioningState
	// contains filtered or unexported fields
}

func NewPartitioningPlan ¶

func NewPartitioningPlan(s state.PartitioningState) PartitioningPlan

func (PartitioningPlan) GetId ¶

func (p PartitioningPlan) GetId() string

type Planner ¶

type Planner interface {
	Plan(ctx context.Context, snapshot Snapshot, pendingPods []v1.Pod) (PartitioningPlan, error)
}

func NewPlanner ¶

func NewPlanner(partitioner PartitionCalculator, sliceCalculator gpu.SliceCalculator, schedulerFramework framework.Framework) Planner

type SliceTracker ¶

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

SliceTracker is a utility struct for tracking the lacking GPU slices of a list of pods

func NewSliceTracker ¶

func NewSliceTracker(snapshot Snapshot, calculator gpu.SliceCalculator, pods []v1.Pod) SliceTracker

func (SliceTracker) GetLackingSlices ¶

func (t SliceTracker) GetLackingSlices() map[gpu.Slice]int

func (SliceTracker) GetRequestedSlices ¶

func (t SliceTracker) GetRequestedSlices() map[gpu.Slice]int

func (SliceTracker) Remove ¶

func (t SliceTracker) Remove(pod v1.Pod)

type Snapshot ¶

type Snapshot interface {
	GetPartitioningState() state.PartitioningState
	GetCandidateNodes() []PartitionableNode
	GetLackingSlices(pod v1.Pod) map[gpu.Slice]int
	SetNode(n PartitionableNode)
	Fork() error
	Commit()
	Revert()
	GetNode(name string) (PartitionableNode, bool)
	GetNodes() map[string]PartitionableNode
	AddPod(node string, pod v1.Pod) error
	Clone() Snapshot
}

func NewClusterSnapshot ¶

func NewClusterSnapshot(
	nodes map[string]PartitionableNode,
	partitioner PartitionCalculator,
	sliceCalculator gpu.SliceCalculator,
	sliceFilter gpu.SliceFilter,
) Snapshot

type SnapshotTaker ¶

type SnapshotTaker interface {
	TakeSnapshot(clusterState *state.ClusterState) (Snapshot, error)
}

type Sorter ¶

type Sorter interface {
	Sort(pods []v1.Pod) []v1.Pod
}

type SorterAdapter ¶

type SorterAdapter func(pods []v1.Pod) []v1.Pod

func NewPodSorter ¶

func NewPodSorter(sliceCalculator gpu.SliceCalculator) SorterAdapter

func (SorterAdapter) Sort ¶

func (f SorterAdapter) Sort(pods []v1.Pod) []v1.Pod

Source Files ¶

View all Source files

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