计算机科学 > 硬件架构
[提交于 2025年8月2日
]
标题: 基于注意力模型部署的千核RV集群自适应共享内存映射的动态分配方案
标题: A Dynamic Allocation Scheme for Adaptive Shared-Memory Mapping on Kilo-core RV Clusters for Attention-Based Model Deployment
摘要: 基于注意力的模型需要灵活的硬件来管理具有不同计算强度和内存访问模式的多样内核。 具有共享L1内存的大规模集群,这是一种常见的架构模式,在扩展时由于层次化PE到L1的集群内互连吞吐量减少,难以充分利用其处理元素(PE)。 本文提出了动态分配方案(DAS),这是一种运行时可编程的地址重映射硬件单元,与统一的内存分配器相结合,旨在最小化PE对多银行L1的数据访问竞争。 我们在一个激进扩展的1024-PE RISC-V集群上评估了DAS,该集群具有非均匀内存访问(NUMA)的PE到L1互连,以展示其在大型并行机器学习工作负载中提高数据局部性的潜力。 对于Vision Transformer(ViT)-L/16模型,每个编码器层执行时间为5.67毫秒,在0.81 PE利用率下,相比固定的字级交错基线实现了1.94倍的速度提升。 在12nm FinFET技术中实现,DAS的面积开销小于0.1%。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.