-
公开(公告)号：CN119166968A
公开(公告)日：2024-12-20
申请号：CN202411205366.2
申请日：2024-08-30
Applicant: 华南理工大学
Abstract: 本发明属于高性能计算技术领域，为基于昇腾Atlas矩阵计算单元的复数矩阵向量乘法计算方法，包含以下步骤：将划分后的计算任务分配给Atlas NPU的各个矩阵计算核心并行计算；vector计算单元将复数向量进行重新组合排布得到实部负虚部间隔和虚部实部间隔的两条列向量，将两条向量组合成一个列优先矩阵；调用矩阵运算单元完成矩阵向量乘法运算，得到复数结果向量的实部向量和虚部向量；通过vector计算单元合并复数结果向量的实部向量和虚部向量，将矩阵运算单元和vector运算单元进行计算核心粒度的同步，得到最终的复数结果向量。本发明无需对复数矩阵进行虚实分离，降低计算性能的消耗，加速复数矩阵向量乘计算，提高了复数矩阵向量乘法在NPU上的计算效率。
Search Results
1
records
(took 0.015 seconds)
筛选
AND
AND
过滤
NOT
NOT
Scan to Join
