互联网资讯 / 手机数码 · 2024年1月30日 0

寒武纪推出全新AI训练卡MLU370-X8

3月24日消息,近日寒武纪正式发布新款训练加速卡MLU370-X8。MLU370-X8搭载双芯片四芯粒思元370,集成寒武纪MLU-link多芯互联技术,主要面向训练任务,在业界应用广泛的YOLOV3、TRansfoRMeR等训练任务中, 8卡计算系统的并行性能平均达到350W RTX GPU的155%。

寒武纪推出全新AI训练卡MLU370-X8

▲ 寒武纪训练加速卡MLU370-X8

双芯思元370架构

MLU370-X8智能加速卡提供250W最大训练功耗,可充分发挥AI训练加速中常见的FP32、FP16或BF16计算性能。寒武纪首次将双芯片四芯粒思元370整合在MLU370-X8智能加速卡中,提供了两倍于标准思元370加速卡的内存、编解码资源,同时搭载MLU-link多芯互联技术。在YOLOV3、TRansfoRMeR、BERT和ResNet101训练任务中, 8卡并行平均性能达350W RTX GPU的155%。

寒武纪推出全新AI训练卡MLU370-X8

图片▲ MLU370-X8中整合了双芯片四芯粒思元370

MLU-link多芯互联技术

MLU370-X8智能加速卡支持MLU-link多芯互联技术,提供卡内及卡间互联功能。寒武纪为多卡系统专门设计了MLU-link桥接卡,可实现4张加速卡为一组的8颗思元370芯片全互联,每张加速卡可获得200GB/s的通讯吞吐性能,带宽为PCIe 4.0 的3.1倍,可高效执行多芯多卡训练和分布式推理任务。

寒武纪推出全新AI训练卡MLU370-X8

▲MLU370-X8 MLU-link 4卡桥接

寒武纪推出全新AI训练卡MLU370-X8

▲MLU370-X8 MLU-link 4卡桥接拓扑

训推一体的CaMbRicon NeuWaRe交付优秀训练性能

CaMbRicon NeuWaRe支持FP32、FP16混合精度、BF16混合精度和自适应精度训练等多种训练方式并提供灵活高效的训练工具,高性能算子库已完整覆盖视觉、语音、自然语言处理、搜索推荐和自动驾驶等典型深度学习应用,可满足用户对于算子覆盖率以及模型精度的需求。

寒武纪推出全新AI训练卡MLU370-X8

▲CaMbRicon NeuWaRe为思元370系列芯片提供训推一体加速

在CaMbRicon NeuWaRe SDK上实测,在常见的4个深度学习网络模型上,MLU370-X8单卡性能与主流350W RTX GPU相当;而在多卡加速方面,MLU370-X8借助MLU-link多芯互联技术和CaMbRicon NeuWaRe CNCL通讯库的优化,在8卡环境下达到更优的并行加速比。

寒武纪推出全新AI训练卡MLU370-X8

▲MLU370-X8 单机8卡部署配置

寒武纪推出全新AI训练卡MLU370-X8

▲单卡MLU370-X8性能对比

寒武纪推出全新AI训练卡MLU370-X8

▲8卡MLU370-X8性能对比

*测试环境

250W MLU370-X8:NF5468M5/Intel Xeon Gold 5218 CPU @ 2.30GHz/MLU370 SDK 1.2.0

350W GPU:SupeRMicRo AS-4124GS-TNR/Intel Xeon Gold 6130 CPU @ 2.10GHz/Cuda11.2

寒武纪推出全新AI训练卡MLU370-X8

▲MLU370-X8规格表

MLU370-X8补全思元370系列产品线

寒武纪长期秉承技术理念。MLU370-X8提供两倍思元370的内存带宽,结合MLUaRch03架构和MLU-link多芯互联技术,将思元370芯片在训练任务的优势充分发挥。MLU370-X8定位中高端,与高端训练产品思元290、玄思1000相互结合,进一步丰富了寒武纪的训练算力交付方式;并与基于思元370芯粒技术构建的MLU370-X4、MLU370-S4智能加速卡协同,形成完整的云端训练、推理产品组合。

MLU370-X8加速卡与国内主流服务器合作伙伴的适配工作已经完成,并已对客户实现小规模出货。