Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type FilterDetail ¶ added in v1.41.0
type FilterRegistry ¶
type FilterRegistry struct {
// contains filtered or unexported fields
}
FilterRegistry provides an immutable collection of GPU filters with methods to create new instances with additional filters
func NewFilterRegistry ¶
func NewFilterRegistry() *FilterRegistry
NewFilterRegistry creates a new empty filter registry
func (*FilterRegistry) Apply ¶
func (fr *FilterRegistry) Apply( ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU, isSimulateSchedule bool, ) ([]tfv1.GPU, []FilterDetail, error)
Apply applies the filters in this registry to the given GPU list Filters are applied in the order they were added (parent filters first)
func (*FilterRegistry) With ¶
func (fr *FilterRegistry) With(filters ...GPUFilter) *FilterRegistry
With creates a new FilterRegistry with the provided filters added The original FilterRegistry is not modified
type GPUFilter ¶
type GPUFilter interface {
// Filter filters the list of GPUs and returns only those that pass the filter criteria
// The implementation should not modify the input slice
Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Name() string
}
GPUFilter defines an interface for filtering GPU candidates
type GPUModelFilter ¶ added in v1.30.0
type GPUModelFilter struct {
// contains filtered or unexported fields
}
GPUModelFilter filters GPUs based on their model (e.g., A100, H100)
func NewGPUModelFilter ¶ added in v1.30.0
func NewGPUModelFilter(model string) *GPUModelFilter
NewGPUModelFilter creates a new filter that matches GPUs with the specified model
func (*GPUModelFilter) Filter ¶ added in v1.30.0
func (f *GPUModelFilter) Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Filter implements GPUFilter interface
func (*GPUModelFilter) Name ¶ added in v1.41.0
func (f *GPUModelFilter) Name() string
type NodeAffinityFilter ¶ added in v1.34.6
type NodeAffinityFilter struct {
// contains filtered or unexported fields
}
func NewNodeAffinityFilter ¶ added in v1.34.6
func NewNodeAffinityFilter(c client.Reader, nodeAffinity *corev1.NodeAffinity) *NodeAffinityFilter
func (*NodeAffinityFilter) Filter ¶ added in v1.34.6
func (f *NodeAffinityFilter) Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Filter
func (*NodeAffinityFilter) Name ¶ added in v1.41.0
func (f *NodeAffinityFilter) Name() string
type PhaseFilter ¶
type PhaseFilter struct {
// contains filtered or unexported fields
}
PhaseFilter filters GPUs based on their operational phase
func NewPhaseFilter ¶
func NewPhaseFilter(allowedPhases ...tfv1.TensorFusionGPUPhase) *PhaseFilter
NewPhaseFilter creates a new PhaseFilter with the specified allowed phases
func (*PhaseFilter) Filter ¶
func (f *PhaseFilter) Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Filter implements GPUFilter.Filter
func (*PhaseFilter) Name ¶ added in v1.41.0
func (f *PhaseFilter) Name() string
type ResourceFilter ¶
type ResourceFilter struct {
// contains filtered or unexported fields
}
ResourceFilter filters GPUs based on available resources
func NewResourceFilter ¶
func NewResourceFilter(required tfv1.Resource) *ResourceFilter
NewResourceFilter creates a new ResourceFilter with the specified resource requirements
func (*ResourceFilter) Filter ¶
func (f *ResourceFilter) Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Filter implements GPUFilter.Filter
func (*ResourceFilter) Name ¶ added in v1.41.0
func (f *ResourceFilter) Name() string
type SameNodeFilter ¶
type SameNodeFilter struct {
// contains filtered or unexported fields
}
SameNodeFilter ensures that the selected GPUs are from the same node
func NewSameNodeFilter ¶
func NewSameNodeFilter(count uint) *SameNodeFilter
NewSameNodeFilter creates a new SameNodeFilter with the specified count
func (*SameNodeFilter) Filter ¶
func (f *SameNodeFilter) Filter(ctx context.Context, workerPodKey tfv1.NameNamespace, gpus []tfv1.GPU) ([]tfv1.GPU, error)
Filter implements GPUFilter.Filter It groups GPUs by node and returns only those nodes that have at least 'count' GPUs
func (*SameNodeFilter) Name ¶ added in v1.41.0
func (f *SameNodeFilter) Name() string
Source Files