Login| Sign Up| Help| Contact|

Patent Searching and Data


Title:
GRAPH EXECUTION PIPELINE PARALLELISM METHOD AND APPARATUS FOR NEURAL NETWORK MODEL COMPUTATION
Document Type and Number:
WIPO Patent Application WO/2023/082575
Kind Code:
A1
Abstract:
Provided in the present invention are a graph execution pipeline parallelism method and apparatus for neural network model computation, and provided are a graph execution pipeline parallelism method and apparatus for neural network model computation in a deep learning training system. The method comprises a graph execution flow during a process for neural network model computation, and a process during which functional modules work cooperatively. According to the graph execution pipeline parallelism method for neural network model computation, graph execution bodies on a local machine are created according to a physical computational graph generated by means of compiling a deep learning framework, and a scheme for allocating a plurality of free memory blocks to each graph execution body is designed, such that the entire computational graph simultaneously participates in deep learning training tasks of different batches of data by means of pipeline parallelism, thereby fully increasing the utilization rate of a memory and the parallelism rate of data.

Inventors:
WANG HONGSHENG (CN)
TAN BOWEN (CN)
BAO HUJUN (CN)
CHEN GUANG (CN)
Application Number:
PCT/CN2022/092481
Publication Date:
May 19, 2023
Filing Date:
May 12, 2022
Export Citation:
Click for automatic bibliography generation   Help
Assignee:
ZHEJIANG LAB (CN)
International Classes:
G06N3/04; G06F12/02
Foreign References:
CN114548383A2022-05-27
CN114237918A2022-03-25
CN114139702A2022-03-04
CN114186687A2022-03-15
CN112884086A2021-06-01
US20190362227A12019-11-28
Attorney, Agent or Firm:
BEIJING ZHILIN HENGYUAN INTELLECTUAL PROPERTY AGENCY CO., LTD. (CN)
Download PDF: