引言
本文介绍SK hynix开发的创新技术:加速器内存(Accelerator-in-Memory,AiM)和AiMX系统。这些先进解决方案旨在应对大型语言模型(LLMs)带来的计算效率和成本效益挑战,适用于数据中心和设备端人工智能场景。
理解挑战:内存受限的大型语言模型
大型语言模型的核心操作在于矩阵-向量乘法,特别是在多头注意力(MHA)和全连接(FC)层中。这些操作主要受内存带宽限制,尤其是在小批量推理时。这一特性使得LLMs的运行成本高昂,无论是在计算资源还是能源消耗方面。
图1:展示了LLM操作的内存受限特性,说明矩阵-向量乘法如何主导计算景观。
SK hynix的解决方案:加速器内存(AiM)
为解决这一挑战,SK hynix推出了加速器内存(AiM)技术。AiM是一种突破性方法,将计算功能更靠近数据,显著减少带宽瓶颈。
GDDR6-AiM的主要特点:
外部带宽:32 GB/s
运行速度:1 GHz
计算吞吐量:512 GFLOPS
内部带宽:512 GB/s
数值精度:BF16
图2:展示了GDDR6-AiM的芯片照片和封装,突出显示了这项创新技术的物理实现。
AiMX:为数据中心应用扩展AiM
在AiM技术的基础上,SK hynix开发了AiMX,这是一种专为数据中心应用设计的系统级解决方案。AiMX卡原型将多个AiM封装与FPGA相结合,创建了一个强大的LLM推理加速器。
AiMX卡原型规格:
主机接口:PCIe Gen3 x8x8(双分叉)
形态因素:FHFL(兼容A100/A30)
配置:2 FPGA x 16 AiM封装
AiM容量:16 GB
带宽:170 GB/s(@2.67Gbps)
图3:展示了AiMX卡原型,演示了多个AiM封装如何集成到单个加速器卡中。
优化AiM中的多头注意力(MHA)
AiM的一个关键优化是其处理多头注意力的方法,这是基于transformer的模型的关键组件。AiM采用智能数据放置策略来处理键和值矩阵,以最大化效率。
图4:说明了AiM感知的键/值矩阵放置策略,展示了数据如何在存储体间分布以实现最佳性能。
扩展AiMX卡以提升性能
SK hynix进一步改进了AiMX解决方案,推出了扩展卡设计,将AiM封装的容量和数量翻倍。
扩展AiMX卡规格:
形态因素:FHFL(兼容H100/A100)
配置:2 FPGA x 32 AiM封装
AiM容量:32 GB
带宽:170 GB/s(@2.67Gbps)
散热:被动式
图5:展示了扩展AiMX卡原型,突出显示了增加的AiM封装数量和改进的设计。
用于设备端人工智能的AiM和AiMX
认识到设备端人工智能的日益重要性,SK hynix将AiM技术适配用于移动和边缘应用。LPDDR-AiM旨在满足设备端人工智能的独特挑战,包括形态因素限制、低批量大小和能效要求。
LPDDR-AiM规格(每个芯片):
内存密度:1-2 GB
组织结构:X16
IO数据率:9.6
外部带宽:19.2 GB/s
数值精度:INT4/8
处理单元:16 PU/芯片
计算吞吐量:307.2 GOPS
内部带宽:153.6 GB/s
图6:展示了LPDDR-AiM的规格,并强调了其与现有LPDDR协议的兼容性。
用于设备端人工智能的AiMX系统架构
设备端AiMX系统架构与其数据中心对应物相似,但已适配移动平台。这种设计允许与移动应用处理器或客户端CPU无缝集成。
图7:比较了数据中心和设备端人工智能应用的AiMX系统架构,说明了相似之处和为移动使用所做的调整。
未来设计考虑
随着SK hynix继续发展AiM和AiMX技术,正在探索几个设计选择:
1. AiM功能:
从GEMV扩展到包括GEMM和其他操作
支持各种精度(二进制、整数、浮点、BF、MX)
探索异构精度能力
2. SoC优化:
平衡功率和热约束
改善MAC到MAC延迟
优化存储体到MAC比率
实施动态电源供应策略
3. 软件增强:
开发PIM感知内存管理技术
实现大页面大小以高效放置权重数据
创建PIM感知内存交换策略
图8:概述了未来AiM实现的各种设计选择,聚焦于功能、精度和潜在应用。
结论和未来展望
SK hynix的AiM和AiMX解决方案在解决内存受限人工智能工作负载的挑战方面代表了重大进步。从数据中心到移动设备,这些技术有望提高性能,降低能源消耗,实现更高效的人工智能推理。
AiM和AiMX的发展路线图雄心勃勃,计划为数据中心提供高容量解决方案,实现设备端人工智能,并探索先进封装技术如混合键合。随着这些技术的成熟,有潜力重塑人工智能加速的格局,使先进的人工智能应用更易获得和更具可持续性。
图9:展示了SK hynix的AiM和AiMX技术路线图,展示了从芯片级解决方案到系统级实现和未来创新的演变。
随着人工智能继续渗透到我们数字生活的各个方面,像AiM和AiMX这样的解决方案将在实现更强大、高效和普遍的人工智能应用中发挥关键作用。研究人员、开发人员和硬件工程师应密切关注这些新兴技术,并考虑如何利用这些技术来推动人工智能的可能性边界。
参考文献
[1]G. Kim et al."SK hynix AI-Specific Computing Memory Solution: From AiM device to Heterogeneous AiMX-xPU System for Comprehensive LLM Inference," in Hot Chips 2024.
点击左下角"阅读原文"马上申请
欢迎转载
转载请注明出处,请勿修改内容和删除作者信息!
关注我们
关于我们:
深圳逍遥科技有限公司(Latitude Design Automation Inc.)是一家专注于半导体芯片设计自动化(EDA)的高科技软件公司。我们自主开发特色工艺芯片设计和仿真软件,提供成熟的设计解决方案如PIC Studio、MEMS Studio和Meta Studio,分别针对光电芯片、微机电系统、超透镜的设计与仿真。我们提供特色工艺的半导体芯片集成电路版图、IP和PDK工程服务,广泛服务于光通讯、光计算、光量子通信和微纳光子器件领域的头部客户。逍遥科技与国内外晶圆代工厂及硅光/MEMS中试线合作,推动特色工艺半导体产业链发展,致力于为客户提供前沿技术与服务。
http://www.latitudeda.com/
(点击上方名片关注我们,发现更多精彩内容)