公开(公告)号：CN115146780B

公开(公告)日：2023-07-11

申请号：CN202211045996.9

申请日：2022-08-30

Applicant: 之江实验室 ,  国家超级计算无锡中心

Inventor： 刘勇 ,  刘鑫 ,  史新民 ,  陈垚键 ,  宋佳伟 ,  王臻 ,  郭楚 ,  付昊桓 ,  高捷 ,  吴汶钊 ,  冯赟龙

IPC: G06N10/20 ,  G06F13/28 ,  G06F9/50

Abstract: 本发明公开一种量子张量网络转置和收缩协同的方法和装置，基于异构众核处理器，包括：步骤一：根据读取的张量信息，进行张量索引预分类后分块读入异构众核处理器的计算处理单元；步骤二：在计算处理单元中，根据索引信息，对读入的分块张量进行转置；步骤三：再对经过转置后的张量依据索信息进行张量收缩；步骤四：收缩完成后，将收缩结果输出回处理器的内存对应张量块位置并合成完整的新收缩后张量结果。本发明通过高效的张量分块、转置和张量收缩计算策略，合理地避免了中间结果的冗余访存开销，实现了转置和张量收缩计算一体化、提高了张量收缩的速度，有效解决了张量收缩计算时间长、效率低的问题，提高了大规模量子电路计算模拟的速度。

公开(公告)号：CN115146780A

公开(公告)日：2022-10-04

申请号：CN202211045996.9

申请日：2022-08-30

Applicant: 之江实验室 ,  国家超级计算无锡中心

Inventor： 刘勇 ,  刘鑫 ,  史新民 ,  陈垚键 ,  宋佳伟 ,  王臻 ,  郭楚 ,  付昊桓 ,  高捷 ,  吴汶钊 ,  冯赟龙

IPC: G06N10/20 ,  G06F13/28 ,  G06F9/50

Abstract: 本发明公开一种量子张量网络转置和收缩协同的方法和装置，基于异构众核处理器，包括：步骤一：根据读取的张量信息，进行张量索引预分类后分块读入异构众核处理器的计算处理单元；步骤二：在计算处理单元中，根据索引信息，对读入的分块张量进行转置；步骤三：再对经过转置后的张量依据索信息进行张量收缩；步骤四：收缩完成后，将收缩结果输出回处理器的内存对应张量块位置并合成完整的新收缩后张量结果。本发明通过高效的张量分块、转置和张量收缩计算策略，合理地避免了中间结果的冗余访存开销，实现了转置和张量收缩计算一体化、提高了张量收缩的速度，有效解决了张量收缩计算时间长、效率低的问题，提高了大规模量子电路计算模拟的速度。