收藏切换
Memory optimization method for control flow computation graph
收藏切换
PDF
Xiangqian WANG1, Yuhao SHEN1, Kun JING1, *, Yafei LYU2
Journal of National Niversity of Defense Technology | 2025, 47(6) : 71 - 80
Less
收藏切换
Journal of National Niversity of Defense Technology | 2025, 47(6): 71-80
Computer System and technology
Memory optimization method for control flow computation graph
Full
Xiangqian WANG1, Yuhao SHEN1, Kun JING1, *, Yafei LYU2
Affiliations
  • 1.School of Internet, Anhui University, Heifei 230039, China
  • 2.iFLYTEK Co., Ltd., Heifei 230026, China
Published: 2025-12-28 doi: 10.11887/j.issn.1001-2486.25050003
Outline
收藏切换

AI chips face on-chip memory limits in deep learning.Current optimization methods focus on static computation graphs, leaving room to improve memory efficiency for dynamic graphs.To overcome this limitation, a memory optimization framework for control-flow computation graphs was developed.The framework realized operator-level memory reuse within subgraphs and further achieved recursive reuse across subgraphs by exploiting control-flow characteristics.In addition, a ping-pong buffering strategy for weight data was introduced to mitigate the memory wall between on-chip and off-chip memory, thereby allowing overlapping of memory access and computation operations within subgraphs.Validation on the domestic LUNA AI chip has demonstrated that the proposed framework improves on-chip memory utilization by 5.9% compared with existing methods.Moreover, the strategy effectively alleviates the memory wall problem by reducing data transfer time between on-chip and off-chip memory, resulting in execution efficiency improvements of up to 29%.

AI chip  /  memory optimization  /  memory reuse  /  cross-memory transfer
Xiangqian WANG, Yuhao SHEN, Kun JING, Yafei LYU. Memory optimization method for control flow computation graph[J]. Journal of National Niversity of Defense Technology, 2025 , 47 (6) : 71 -80 . DOI: 10.11887/j.issn.1001-2486.25050003
Year 2025 volume 47 Issue 6
PDF
56
24
Cite this Article
BibTeX
Article Info
doi: 10.11887/j.issn.1001-2486.25050003
  • Receive Date:2025-05-06
  • Online Date:2026-04-16
  • Published:2025-12-28
Article Data
Affiliations
History
  • Received:2025-05-06
Affiliations
    1.School of Internet, Anhui University, Heifei 230039, China
    2.iFLYTEK Co., Ltd., Heifei 230026, China
References
Share
https://castjournals.cast.org.cn/joweb/gfkjdxxb/EN/10.11887/j.issn.1001-2486.25050003
Share to
QR

Scan QR to access full text

Cite this article
BibTeX
Citations
表12种不同金属材料的力学参数

Family
属数
Number of
genus
种数
Number of
species
占总种数比例
Percentage of
total species (%)

Genus
种数
Number of
species
占总种数比例
Percentage of total
species (%)
鹅膏菌科Amanitaceae 2 11 5.26 鹅膏菌属 Amanita 10 4.78
小菇科 Mycenaceae 2 12 5.74 丝盖伞属 Inocybe 5 2.39
多孔菌科 Polyporaceae 8 14 6.70 蜡蘑属 Laccaria 5 2.39
红菇科 Russulaceae 3 23 11.00 小皮伞属 Marasmius 6 2.87
小菇属 Mycena 11 5.26
光柄菇属 Pluteus 5 2.39
红菇属 Russula 17 8.13
栓菌属 Trametes 5 2.39
关闭全屏
  • BibTeX
  • EndNote
  • RefWorks
  • TxT