Login| Sign Up| Help| Contact|

Patent Searching and Data


Title:
ADAPTATION METHOD AND APPARATUS FOR DISTRIBUTED TRAINING IN DEEP LEARNING FRAMEWORK AND AI ACCELERATION CARD
Document Type and Number:
WIPO Patent Application WO/2023/103122
Kind Code:
A1
Abstract:
Disclosed in the present invention are an adaptation method and apparatus for distributed training in deep learning framework and AI acceleration card. The method comprises the following steps: S1: a deep learning framework supporting single-card configuration in a newly-added AI acceleration card, the sub-steps comprising: S11: the deep learning framework supporting new hardware; S12, the deep learning framework supporting device threads of the new hardware; S13: the deep learning framework supporting memory operation of the new hardware; and S14: the deep learning framework supporting an operator kernel function of the new hardware; S2: the deep learning framework supporting multi-card configuration in the newly-added AI acceleration card; S3: the deep learning framework supporting tensor segmentation and multi-card distribution; and S4: the deep learning framework supports multi-card set communication in the newly-added AI acceleration card. In the present invention, a relationship between the deep learning framework and the newly-added AI acceleration card is established, the deep learning framework source code is comprehensively introduced to the chip bottom layer software, the hardware capability of the chip is released to the maximum extent, providing the strongest computing power for the end side AI.

Inventors:
WANG HONGSHENG (CN)
HUA WEI (CN)
BAO HUJUN (CN)
Application Number:
PCT/CN2021/142586
Publication Date:
June 15, 2023
Filing Date:
December 29, 2021
Export Citation:
Click for automatic bibliography generation   Help
Assignee:
ZHEJIANG LAB (CN)
International Classes:
G06F9/50
Foreign References:
CN109460827A2019-03-12
CN113342346A2021-09-03
CN113076143A2021-07-06
US20200250525A12020-08-06
CN113918351A2022-01-11
Attorney, Agent or Firm:
HANGZHOU QIUSHI PATENT OFFICE CO., LTD. (CN)
Download PDF: