可与英伟达(NVIDIA)相媲美的高容量、高性能人工智能(AI)加速器已在韩国本土开发完成。当前英伟达几乎垄断了AI加速器市场。韩国本土研究团队通过搭载下一代接口技术的高容量、高性能AI加速器,向英伟达的一家独大格局发起挑战。


(从上排左侧起顺时针方向)电气及电子工程学部硕士研究生 Kim Dongpyeong、硕士研究生 Yu Eojin、博士 Lee Sangwon、博士研究生 Guk Donghyeon、教授 Jeong Myeongsu、博士研究生 Kang Seungkwan、博士研究生 Jang Junhyeok、博士研究生 Bae Hanyeoreum。KAIST 提供

(从上排左侧起顺时针方向)电气及电子工程学部硕士研究生 Kim Dongpyeong、硕士研究生 Yu Eojin、博士 Lee Sangwon、博士研究生 Guk Donghyeon、教授 Jeong Myeongsu、博士研究生 Kang Seungkwan、博士研究生 Jang Junhyeok、博士研究生 Bae Hanyeoreum。KAIST 提供

View original image

韩国科学技术院(KAIST)8日表示,电气及电子工程系Jung Myungsoo教授研究团队(计算机体系结构及存储系统实验室)已开发出一项技术,通过激活下一代接口技术CXL(Compute Express Link),优化大容量图形处理器(GPU)内存读写性能。


最新GPU的内部内存容量仅为数十千兆字节(GB),单凭一块GPU无法完成模型推理和训练。出于同样原因,业界为满足大规模AI模型对内存容量的需求,普遍采用多块GPU互连的方式,但鉴于GPU市场价格高企,这种方式会大幅推高总拥有成本(TCO,Total Cost of Ownership)。


产业界之所以积极研究通过下一代互连技术“CXL”将大容量内存直接连接到GPU设备上的“CXL-GPU”架构技术,正是出于这一考虑。


CXL-GPU通过将CXL互连的内存扩展设备的内存空间整合进GPU内存空间,从而实现大容量支持。整合后内存空间管理所需的操作由CXL控制器自动完成,GPU可以用与访问原有本地内存相同的方式访问扩展内存空间。与以往为扩充内存容量而追加采购高价GPU的做法不同,CXL-GPU只需为GPU有选择性地增加内存资源即可,因而在系统构建成本上具有大幅削减的优势。


然而,仅凭CXL-GPU的大容量特性,还难以真正应用于AI服务。由于大规模AI服务需要极快的推理和训练性能,只有当直接连接至GPU的内存扩展设备在内存读写方面能提供接近GPU本地内存的性能时,才能真正用于AI服务。


研究团队此次开发的技术,通过分析并改善CXL-GPU设备内存读写性能下降的原因,具有重要意义。团队开发出一种允许内存扩展设备自主决定内存写入时机的技术:当GPU设备向内存扩展设备发出内存写入请求时,系统同时在GPU本地内存中执行写入操作。


通过根据内存扩展设备的内部工作状态安排操作,GPU无需等待直至确认内存写入任务完成,从而解决了写入性能下降的问题。


研究团队还开发了这样一项技术:由GPU设备预先向内存扩展设备提供提示,使其能够提前执行内存读取。利用这一技术,内存扩展设备可以更早开始内存读取,当GPU设备真正需要数据时,即可从缓存(小而快速的临时数据存储空间)中读取,从而实现更高的内存读取性能。


本次研究利用了半导体无晶圆厂初创企业Panmnesia的超高速CXL控制器和CXL-GPU原型机开展。


CXL-GPU 示意图片。KAIST 提供

CXL-GPU 示意图片。KAIST 提供

View original image

Panmnesia拥有业内首款将CXL内存管理操作往返时延压缩至两位数纳秒(nanosecond)以下的自研CXL控制器。该控制器完全由韩国本土技术开发而成,运行速度比世界各国推出的最新CXL控制器快3倍以上。


Panmnesia利用高速CXL控制器,将多台内存扩展设备直接连接到GPU,使单块GPU即可构成太字节(TB)级的大规模内存空间。


研究团队基于Panmnesia的CXL-GPU原型机进行技术有效性验证,确认其运行AI服务的速度比既有GPU内存扩展技术快2.36倍。


研究团队的研究成果将于本月在圣克拉拉举行的USENIX联合学会及HotStorage研究发表会上公开。


Jung Myungsoo教授表示:“大规模语言模型在训练时需要数至数十太字节的内存,处于AI服务提供最前沿的大型科技企业为满足这一需求,正竞相扩大模型和数据规模。我们认为,此次研究的意义在于,团队开发出了搭载下一代接口技术、能够对抗当前垄断AI加速器市场的英伟达的高容量、高性能AI加速器。”



他还补充称:“我们期待此次开发的技术能够加速CXL-GPU市场的启动进程,并有望大幅降低运营大规模AI服务的大型科技企业的内存扩展成本。”


本报道由人工智能(AI)翻译技术生成。

版权所有 © 阿视亚经济。未经许可不得转载,禁止用于AI训练及使用。