internal

package

v0.0.0-...-c70b2fc Latest Latest Go to latest Published: Apr 23, 2025 License: Apache-2.0, Apache-2.0 Imports: 8 Imported by: 0

Details

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

Repository

github.com/professorshandian/mind-cluster

Links

Open Source Insights

Documentation ¶

Overview ¶

Package internal is using for HuaWei Ascend pin scheduling policy schedule.

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func New ¶

func New() plugin.SchedulerPluginNeed

New new scheduler driver controller

Types ¶

type Controller ¶

type Controller struct {
	PolicyHandler []plugin.SchedulerPluginNeed
}

Controller controller to driver the scheduling process

func (*Controller) CheckNodeNPUByTask ¶

func (c *Controller) CheckNodeNPUByTask(task *api.TaskInfo, node plugin.NPUNode) error

CheckNodeNPUByTask check nod npu meet task req

func (*Controller) InitMyJobPlugin ¶

func (c *Controller) InitMyJobPlugin(attr util.SchedulerJobAttr, env plugin.ScheduleEnv) error

InitMyJobPlugin set attr and env for plugin

func (*Controller) PreStartAction ¶

func (c *Controller) PreStartAction(ssn *framework.Session) error

PreStartAction pre-processing actions for all policy handler

func (*Controller) ReleaseAnnotation ¶

func (c *Controller) ReleaseAnnotation(task *api.TaskInfo, node plugin.NPUNode) *plugin.NPUNode

ReleaseAnnotation release annotation

func (*Controller) ScoreBestNPUNodes ¶

func (c *Controller) ScoreBestNPUNodes(task *api.TaskInfo, nodes []*api.NodeInfo, sMap map[string]float64) error

ScoreBestNPUNodes score node by calculate task req npu num and node npu top

func (*Controller) SetPolicyHandler ¶

func (c *Controller) SetPolicyHandler(attr util.SchedulerJobAttr, env plugin.ScheduleEnv)

SetPolicyHandler set attr and env for plugin

func (*Controller) UseAnnotation ¶

func (c *Controller) UseAnnotation(task *api.TaskInfo, node plugin.NPUNode) *plugin.NPUNode

UseAnnotation select npu for task from node

func (*Controller) ValidNPUJob ¶

func (c *Controller) ValidNPUJob() *api.ValidateResult

ValidNPUJob check job req npu num

Source Files ¶

View all Source files

controller.go

Directories ¶

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

Path	Synopsis
npu Package npu is using for HuaWei Ascend pin affinity schedule.	Package npu is using for HuaWei Ascend pin affinity schedule.
ascend310/card310x4 Package card310x4 is using for HuaWei A300T Ascend pin affinity schedule.	Package card310x4 is using for HuaWei A300T Ascend pin affinity schedule.
ascend310/chip310x4 Package chip310x4 is using for HuaWei 310 Ascend pin affinity schedule.	Package chip310x4 is using for HuaWei 310 Ascend pin affinity schedule.
ascend310p/card310px2 Package card310px2 is using for HuaWei 300I Duo Ascend pin affinity schedule.	Package card310px2 is using for HuaWei 300I Duo Ascend pin affinity schedule.
ascend310p/chip310px2 Package chip310px2 is using for HuaWei 300I Duo Ascend pin affinity schedule.	Package chip310px2 is using for HuaWei 300I Duo Ascend pin affinity schedule.
ascend310p/vnpu Package vnpu is using for Ascend vnpu affinity schedule.	Package vnpu is using for Ascend vnpu affinity schedule.
ascend910/ascend910a3 Package ascend910a3 is using for A3 affinity schedule.	Package ascend910a3 is using for A3 affinity schedule.
ascend910/ascend910a3/module910a3x16 Package module910a3x16 is using for A3 x16 affinity schedule.	Package module910a3x16 is using for A3 x16 affinity schedule.
ascend910/ascend910a3/superpod Package superpod is using for HuaWei Atlas 900 A3 SuperPod affinity schedule.	Package superpod is using for HuaWei Atlas 900 A3 SuperPod affinity schedule.
ascend910/ascend910b Package ascend910b is using for HuaWei Ascend 910B pin affinity schedule.	Package ascend910b is using for HuaWei Ascend 910B pin affinity schedule.
ascend910/ascend910b/module910bx16 Package module910bx16 is using for HuaWei Ascend910B A+X pin affinity schedule.	Package module910bx16 is using for HuaWei Ascend910B A+X pin affinity schedule.
ascend910/ascend910b/vnpu Package vnpu is using for Ascend vnpu affinity schedule.	Package vnpu is using for Ascend vnpu affinity schedule.
ascend910/ascend910old/module910x8 Package module910x8 is using for HuaWei A800/9000 Ascend910 pin affinity schedule.	Package module910x8 is using for HuaWei A800/9000 Ascend910 pin affinity schedule.
base Package base is using for HuaWei Ascend pin affinity schedule.	Package base is using for HuaWei Ascend pin affinity schedule.
vnpu Package vnpu is using for HuaWei Ascend pin vnpu allocation.	Package vnpu is using for HuaWei Ascend pin vnpu allocation.
nslb Package nslb is using for HuaWei Ascend pin tor affinity.	Package nslb is using for HuaWei Ascend pin tor affinity.
rescheduling Package rescheduling is using for HuaWei Ascend pin fault rescheduling.	Package rescheduling is using for HuaWei Ascend pin fault rescheduling.
test Package test is using for HuaWei Ascend testing.	Package test is using for HuaWei Ascend testing.