pod_info

package

v0.9.13 Latest Latest Go to latest Published: Jan 26, 2026 License: Apache-2.0 Imports: 18 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 ¶

Constants
func IsKaiUtilityPod(pod *v1.Pod) bool
func IsResourceReservationTask(pod *v1.Pod) bool
func IsScaleAdjustTask(pod *v1.Pod) bool
func PodKey(pod *v1.Pod) common_info.PodID
type PVCInfo
type PodInfo
- func NewTaskInfo(pod *v1.Pod) *PodInfo
- func NewTaskInfoWithBindRequest(pod *v1.Pod, bindRequest *bindrequest_info.BindRequestInfo) *PodInfo
type PodStorageRequirements
type PodsMap
type ResourceReceivedType
type ResourceRequestType

Constants ¶

View Source

const (
	GpuMemoryAnnotationName            = "gpu-memory"
	GPUGroup                           = "runai-gpu-group"
	ReceivedResourceTypeAnnotationName = "received-resource-type"
	WholeGpuIndicator                  = "-2"
)

Variables ¶

This section is empty.

Functions ¶

func IsKaiUtilityPod ¶ added in v0.5.0

func IsKaiUtilityPod(pod *v1.Pod) bool

func IsResourceReservationTask ¶ added in v0.5.0

func IsResourceReservationTask(pod *v1.Pod) bool

func IsScaleAdjustTask ¶ added in v0.5.0

func IsScaleAdjustTask(pod *v1.Pod) bool

func PodKey ¶

func PodKey(pod *v1.Pod) common_info.PodID

PodKey returns the string key of a pod.

Types ¶

type PVCInfo ¶

type PVCInfo struct {
	StorageClassName string                        `json:"storageClassName"`
	Size             *resource.Quantity            `json:"size"`
	Phase            v1.PersistentVolumeClaimPhase `json:"phase"`
}

type PodInfo ¶

type PodInfo struct {
	UID common_info.PodID
	Job common_info.PodGroupID

	Name      string
	Namespace string

	SubGroupName string

	ResourceRequestType  ResourceRequestType
	ResourceReceivedType ResourceReceivedType

	// ResReq are the minimal resources that needed to launch a pod. (includes init containers resources)
	ResReq           *resource_info.ResourceRequirements
	AcceptedResource *resource_info.ResourceRequirements

	GPUGroups []string

	NodeName        string
	Status          pod_status.PodStatus
	IsVirtualStatus bool
	IsLegacyMIGtask bool

	BindRequest *bindrequest_info.BindRequestInfo

	ResourceClaimInfo bindrequest_info.ResourceClaimInfo

	Pod *v1.Pod
	// contains filtered or unexported fields
}

func NewTaskInfo ¶

func NewTaskInfo(pod *v1.Pod) *PodInfo

func NewTaskInfoWithBindRequest ¶

func NewTaskInfoWithBindRequest(pod *v1.Pod, bindRequest *bindrequest_info.BindRequestInfo) *PodInfo

func (*PodInfo) Clone ¶

func (pi *PodInfo) Clone() *PodInfo

func (*PodInfo) GetAllStorageClaims ¶

func (pi *PodInfo) GetAllStorageClaims() map[storageclaim_info.Key]*storageclaim_info.StorageClaimInfo

func (*PodInfo) GetDeletedStorageClaimsNames ¶

func (pi *PodInfo) GetDeletedStorageClaimsNames() string

func (*PodInfo) GetOwnedStorageClaims ¶

func (pi *PodInfo) GetOwnedStorageClaims() map[storageclaim_info.Key]*storageclaim_info.StorageClaimInfo

func (*PodInfo) GetSchedulingConstraintsSignature ¶

func (pi *PodInfo) GetSchedulingConstraintsSignature() common_info.SchedulingConstraintsSignature

func (*PodInfo) GetUnboundOrReleasingStorageClaimsByStorageClass ¶

func (pi *PodInfo) GetUnboundOrReleasingStorageClaimsByStorageClass() map[common_info.StorageClassID][]*storageclaim_info.StorageClaimInfo

func (*PodInfo) IsCPUOnlyRequest ¶

func (pi *PodInfo) IsCPUOnlyRequest() bool

func (*PodInfo) IsFractionAllocation ¶

func (pi *PodInfo) IsFractionAllocation() bool

func (*PodInfo) IsFractionCandidate ¶

func (pi *PodInfo) IsFractionCandidate() bool

func (*PodInfo) IsFractionRequest ¶

func (pi *PodInfo) IsFractionRequest() bool

func (*PodInfo) IsMemoryRequest ¶

func (pi *PodInfo) IsMemoryRequest() bool

func (*PodInfo) IsMigCandidate ¶

func (pi *PodInfo) IsMigCandidate() bool

func (*PodInfo) IsMigProfileAllocation ¶

func (pi *PodInfo) IsMigProfileAllocation() bool

func (*PodInfo) IsMigProfileRequest ¶

func (pi *PodInfo) IsMigProfileRequest() bool

func (*PodInfo) IsRegularGPURequest ¶

func (pi *PodInfo) IsRegularGPURequest() bool

func (*PodInfo) IsRequireAnyKindOfGPU ¶

func (pi *PodInfo) IsRequireAnyKindOfGPU() bool

func (*PodInfo) IsSharedGPUAllocation ¶

func (pi *PodInfo) IsSharedGPUAllocation() bool

func (*PodInfo) IsSharedGPURequest ¶

func (pi *PodInfo) IsSharedGPURequest() bool

func (*PodInfo) ShouldAllocate ¶

func (pi *PodInfo) ShouldAllocate(isRealAllocation bool) bool

func (PodInfo) String ¶

func (pi PodInfo) String() string

func (*PodInfo) UpsertStorageClaim ¶

func (pi *PodInfo) UpsertStorageClaim(claimInfo *storageclaim_info.StorageClaimInfo)

type PodStorageRequirements ¶

type PodStorageRequirements struct {
	PVCs         []PVCInfo   `json:"pvcs"`
	OtherVolumes []v1.Volume `json:"otherVolumes"`
}

type PodsMap ¶

type PodsMap map[common_info.PodID]*PodInfo

type ResourceReceivedType ¶

type ResourceReceivedType string

const (
	ReceivedTypeMigInstance ResourceReceivedType = "MigInstance"
	ReceivedTypeFraction    ResourceReceivedType = "Fraction"
	ReceivedTypeRegular     ResourceReceivedType = "Regular"
	ReceivedTypeNone        ResourceReceivedType = ""
)

type ResourceRequestType ¶

type ResourceRequestType string

const (
	RequestTypeGpuMemory   ResourceRequestType = "GpuMemory"
	RequestTypeMigInstance ResourceRequestType = "MigInstance"
	RequestTypeFraction    ResourceRequestType = "Fraction"
	RequestTypeRegular     ResourceRequestType = "Regular"
)

Source Files ¶

View all Source files

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