Operator fusion method and device, equipment and storage medium

• Main Authors: ZHENG HUIHUANG, CHEN TEFENG, CHEN HAOZE, SUN LI, SHI XIAOWEI, WANG ZHEN, LAN XIANG, LIU YIQUN, JIANG CHENG
• Format: Patent
• Language: Chinese; English
• Published: 14.03.2023
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; PHYSICS

The invention provides an operator fusion method and device, equipment and a storage medium, and relates to the technical field of artificial intelligence, in particular to the technical field of deep learning and the like. The method comprises the following steps: acquiring an operator set of a neural network model; performing topological sorting on each operator in the operator set to obtain a first sorting result; fusing the operators based on the first sorting result and a preset operator fusion rule to obtain a first fusion result, the first fusion result comprising at least one fused sub-graph; performing topological sorting on each sub-graph in the first fusion result to obtain a second sorting result; and fusing the sub-graphs based on the second sorting result and a preset sub-graph fusion rule to obtain a second fusion result. According to the operator fusion method provided by the invention, the operator fusion efficiency is improved, and the number of fused sub-graphs is reduced, so that the calcu