- 专利标题: MEMORY ALLOCATION METHOD FOR SPARSE MATRIX MULTIPLICATION APPLICATIONS
-
申请号: US18309826申请日: 2023-04-30
-
公开(公告)号: US20240184848A1公开(公告)日: 2024-06-06
- 发明人: Zhaoyang DU , Yijin GUAN , Dimin NIU , Tianchan GUAN , Hongzhong ZHENG
- 申请人: Alibaba (China) Co., Ltd.
- 申请人地址: CN Hangzhou
- 专利权人: Alibaba (China) Co., Ltd.
- 当前专利权人: Alibaba (China) Co., Ltd.
- 当前专利权人地址: CN Hangzhou
- 优先权: CN 2211535033.7 2022.12.02
- 主分类号: G06F17/16
- IPC分类号: G06F17/16 ; G06F7/487
摘要:
This application describes accelerator, computer system, and method for memory allocation in sparse matrix-matrix multiplications (spGEMM). An example method may include: computing a number of floating point multiplication operations (FLOP) to be performed to generate each row in the output matrix; determining an estimated compression ratio based on a plurality of first rows sampled from the first sparse matrix and a plurality of corresponding second rows from the second sparse matrix; determining an estimated number of non-zero data (NNZ) in each row of the to-be-generated output matrix; constructing a plurality of hash tables for the rows in the to-be-generated output matrix based on the estimated NNZ corresponding to each row; performing symbolic computations between the first and second sparse matrices by using the hash tables to determine actual NNZs in the to-be-generated output matrix; and allocating a memory space for the output matrix based on the actual NNZs.
信息查询