NEURAL DIRECTED ACYCLIC GRAPH (DAG) SCHEDULING VIA ONE-SHOT PRIORITY SAMPLING

    公开(公告)号:US20240119301A1

    公开(公告)日:2024-04-11

    申请号:US18464996

    申请日:2023-09-11

    CPC classification number: G06N3/092

    Abstract: A processor-implemented method includes sampling, according to a priority sampling policy, a set of node priorities from a computation graph. Each node priority of the set of node priorities may be associated with a respective node on the computation graph. Additionally, each node may represent an operation of a task performed by an artificial neural network. The method also includes converting, via a list scheduling function, the node priorities to a schedule that associates each node of the computation graph with a processor of a group of processors of a device associated with the artificial neural network, the schedule associated with a makespan. The method further includes performing the task in accordance with the schedule.

Patent Agency Ranking