ascend910b

Documentation

Overview

Package ascend910b is using for HuaWei Ascend 910B pin affinity schedule.

Package ascend910b is using for HuaWei Ascend 910B pin affinity schedule.

Package ascend910b is using for HuaWei Ascend 910B pin affinity schedule.

Package ascend910b is using for HuaWei Ascend pin affinity schedule.

Package ascend910b is using for HuaWei pin affinity schedule.

Index

• func GetVNPUTaskCpuLevel(asTask util.NPUTask) string
• func GetVNPUTaskDVPP(asTask util.NPUTask) string
• type Base910b
  • func (ab *Base910b) Check910bNodeNPUByTask(task *api.TaskInfo, node plugin.NPUNode) error
  • func (ab *Base910b) CheckJobAllowNum(value int) bool
  • func (ab *Base910b) CheckJobForm() error
  • func (ab *Base910b) GetAcceleratorValue() string
  • func (ab *Base910b) GetArch() string
  • func (ab *Base910b) GetNPUAllocPriorityArray(taskNPUNumber int) ([]int, error)
  • func (ab *Base910b) GetNodeBestScore(taskNPUNum int, npuTop []int) (int, error)
  • func (ab *Base910b) GetNodeHccsArray(nodeTop []int) ([]int, []int)
  • func (tp *Base910b) GetPresetVirtualDevices()
  • func (tp *Base910b) GetVNPUTemplate()
  • func (tp *Base910b) InitVNPU()
  • func (ab *Base910b) Judge910BNodeAndTaskNPU(taskNPU int, nodeTop []int) error
  • func (tp *Base910b) PreStartVNPU(ssn *framework.Session) error
  • func (ab *Base910b) ScoreAscendNPUNodes(task *api.TaskInfo, nodes []*api.NodeInfo, sMap map[string]float64) error
  • func (ab *Base910b) SetAcceleratorValue(value string)
  • func (ab *Base910b) SetArch(value string)
  • func (ab *Base910b) SetNpuNumInvalidMap(value map[int]struct{})
  • func (ab *Base910b) UpdateNodeInfo(node plugin.NPUNode, usedTop []int) *plugin.NPUNode
  • func (ab *Base910b) Use910bAnnotation(task *api.TaskInfo, node plugin.NPUNode) *plugin.NPUNode
  • func (ab *Base910b) Valid910bNPUJob() *api.ValidateResult
  • func (tp *Base910b) ValidDyVNPUJob() *api.ValidateResult
• type SelectNodeInf

Functions

func GetVNPUTaskCpuLevel

func GetVNPUTaskCpuLevel(asTask util.NPUTask) string

GetVNPUTaskCpuLevel cpu default is null

func GetVNPUTaskDVPP

func GetVNPUTaskDVPP(asTask util.NPUTask) string

GetVNPUTaskDVPP dvpp default is null

Types

type Base910b

type Base910b struct {
	base.NPUHandler
	VHandle          *vnpu.VirtualNPU
	AffScoreList     [][]int
	NpuNumInvalidMap map[int]struct{}
	// contains filtered or unexported fields
}

Base910b for Ascend 910B base.

func (*Base910b) Check910bNodeNPUByTask

func (ab *Base910b) Check910bNodeNPUByTask(task *api.TaskInfo, node plugin.NPUNode) error

Check910bNodeNPUByTask check nod npu meet 910B task req

func (*Base910b) CheckJobAllowNum

func (ab *Base910b) CheckJobAllowNum(value int) bool

CheckJobAllowNum check the single job require is valid. eg:A+X 16P:1,2,4,8,16;A+K 1,2,4,8.

func (*Base910b) CheckJobForm

func (ab *Base910b) CheckJobForm() error

CheckJobForm to check job ring-controller.atlas for future unification.

func (*Base910b) GetAcceleratorValue

func (ab *Base910b) GetAcceleratorValue() string

GetAcceleratorValue Get the acceleratorValue to distinguish between task types.

func (*Base910b) GetArch

func (ab *Base910b) GetArch() string

GetArch Get the job arch to distinguish between jobs. A+X 16P,A+K 8p.

func (*Base910b) GetNPUAllocPriorityArray

func (ab *Base910b) GetNPUAllocPriorityArray(taskNPUNumber int) ([]int, error)

GetNPUAllocPriorityArray get priorityArray

func (*Base910b) GetNodeBestScore

func (ab *Base910b) GetNodeBestScore(taskNPUNum int, npuTop []int) (int, error)

GetNodeBestScore Get node core

func (*Base910b) GetNodeHccsArray

func (ab *Base910b) GetNodeHccsArray(nodeTop []int) ([]int, []int)

GetNodeHccsArray get node hccs array

func (*Base910b) GetPresetVirtualDevices

func (tp *Base910b) GetPresetVirtualDevices()

GetPresetVirtualDevices get preset virtual devices

func (*Base910b) GetVNPUTemplate

func (tp *Base910b) GetVNPUTemplate()

GetVNPUTemplate get vnpu template

func (*Base910b) InitVNPU

func (tp *Base910b) InitVNPU()

InitVNPU init map of vnpu tmp

func (*Base910b) Judge910BNodeAndTaskNPU

func (ab *Base910b) Judge910BNodeAndTaskNPU(taskNPU int, nodeTop []int) error

Judge910BNodeAndTaskNPU Judge 910BNode wither meet npu task not.

func (*Base910b) PreStartVNPU

func (tp *Base910b) PreStartVNPU(ssn *framework.Session) error

PreStartVNPU do something before schedule for vnpu

func (*Base910b) ScoreAscendNPUNodes

func (ab *Base910b) ScoreAscendNPUNodes(task *api.TaskInfo, nodes []*api.NodeInfo, sMap map[string]float64) error

ScoreAscendNPUNodes core ascend910B node by calculate task req npu num and node npu top

func (*Base910b) SetAcceleratorValue

func (ab *Base910b) SetAcceleratorValue(value string)

SetAcceleratorValue Set the acceleratorValue to distinguish between task types.

func (*Base910b) SetArch

func (ab *Base910b) SetArch(value string)

SetArch Set the job arch to distinguish between jobs. A+X 16P,A+K 8p.

func (*Base910b) SetNpuNumInvalidMap

func (ab *Base910b) SetNpuNumInvalidMap(value map[int]struct{})

SetNpuNumInvalidMap Set the single job not allow number. eg:A+X 16P:9,10,11,12,13,14,15

func (*Base910b) UpdateNodeInfo

func (ab *Base910b) UpdateNodeInfo(node plugin.NPUNode, usedTop []int) *plugin.NPUNode

UpdateNodeInfo update node info

func (*Base910b) Use910bAnnotation

func (ab *Base910b) Use910bAnnotation(task *api.TaskInfo, node plugin.NPUNode) *plugin.NPUNode

Use910bAnnotation select npu for 910b task from node

func (*Base910b) Valid910bNPUJob

func (ab *Base910b) Valid910bNPUJob() *api.ValidateResult

Valid910bNPUJob check the 910b job req npu num and mode

func (*Base910b) ValidDyVNPUJob

func (tp *Base910b) ValidDyVNPUJob() *api.ValidateResult

ValidDyVNPUJob valid dynamic cut job

type SelectNodeInf

type SelectNodeInf struct {
	AllNPUNum   int
	LeftNPUNum  int
	RightNPUNum int
	// contains filtered or unexported fields
}

SelectNodeInf for node hccs.