壁仞科技支持DeepSeek
DeepSeek在开源周开源了部分关键模块的壁仞代码及推理系统参考架构,再次引发行业震动,科技但目前尚未开源DeepSeek-V3 满血版完整训练代码。支持壁仞科技凭借八大自主创新技术,壁仞实现DeepSeek-V3满血版在国产GPU平台的科技高效全栈式训练与推理,实现国产大模型与国产GPU的支持深度融合优化,开启国产算力新篇章。壁仞
01.首个国产化Megatron-LM
深度适配方案
DeepSeek业界首次提出了Multi-headLatent Attention (MLA) 、科技Multi-Token Prediction (MTP)、支持Auxiliary-Loss-Free Load Balancing等新特性,壁仞但并未开放训练代码,科技且Megatron-LM初期也未支持DeepSeek-V3。支持壁仞科技充分利用自研的壁仞Megatron-LM-BR插件、训练框架BR-PyTorch以及高性能加速库suDNN、科技suBLAS、支持suCCL等,在Megatron-LM框架上率先支持了这些新特性。针对DeepSeek-V3预训练,壁仞支持四大并行策略(TP/EP/PP/DP)灵活组合,较原生方案(TP1PP16EP64DP2)更容易适应不同规模节点的灵活扩展。实现"Attention-MoE异构TP"技术,大幅降低了Attention显存需求。
02.突破海外GPU高度依赖,
实现关键技术国产替代
DeepSeek基于海外GPU做了大量算法和软硬件协同优化,其实现高度依赖海外GPU微架构、NVLink互连、IB交换机等。基于自研BIREN大算力TCore、高速互连BLink等技术,壁仞科技在DeepSeek开源周之前就已经实现了FlashMLA、DeepGEMM、DeepEP等核心模块类似功能和优化技术,并进行了软硬协同深度优化,端到端打通了训练全流程,实现了关键技术的国产自主可控。
03.业界首创显存优化双擎技术,
实现满血版高效训练
DeepSeek-V3满血版有671B参数,官方推荐方案至少需要1024个H800(TP1PP16EP64DP1)才能进行全参训练。为了应对显存挑战,壁仞科技自主创新Async Offload、GPU-based Chunk Optimizer等精度无损的显存优化技术,将大量激活值及优化器状态异步卸载至CPU内存,实现仅需4096GB显存(8-16台单机8卡GPU机器)即可支持DeepSeek-V3全参高效训练。与此同时,通过巧妙地重叠计算与通信,上述显存优化技术几乎没有带来训练性能的损失。另外支持智能重计算策略,自动识别显存瓶颈层,实现"算力换空间"智能决策。通过Async Offload和重计算显存优化双擎技术,实现算力开销和显存节约的最佳均衡。
04.业界首创"虚拟层+动态重排"技术,
大幅降低流水线气泡
DeepSeek的DualPipe技术能掩盖部分传输开销及大幅降低流水线气泡,但DualPipe需要GPU微架构层面对计算单元进行精确控制才能实现计算和通信的高效并发,相比1F1B,DualPipe需要增加1倍的模型权重及部分激活值显存消耗。相比1F1B,Interleave with Virtual Pipeline技术可以在不增加显存消耗的情况下大幅降低流水线气泡,但是要求每个PP Stage的layer数是均衡的。DeepSeek-V3有61个Layer,PP8/PP16都无法均衡切分Layer导致无法使能Virtual Pipeline,壁仞科技业界首创"虚拟层+动态重排"技术,通过插入3个虚拟层实现64层均匀切分,重排部分层实现负载均衡,避免流水线等待,最终实现高效PP并行。
05.融合算子加速体系,
释放国产GPU性能潜力
针对DeepSeek-V3的Linear、Attention、YaRN RoPE、Dispatch、Combine等关键耗时算子,壁仞科技基于其GPU架构特点做了极致的图算/通算融合优化,从多计算操作极致的片上融合,张量处理器与矢量处理器极致异步融合,多级缓存的流水融合,到计算与通信融合,并进一步引入自动化的Kernel Selection技术,基于硬件计算/通信/访存建模的CostModel针对不同工作负载自适应选择最优内核实现,将芯片综合能效发挥到极致。
06.整合关键技术模块,实现DeepSeek-
V3满血版端到端高效预训练
壁仞科技壁砺TM系列产品如壁砺TM106M、106B等产品具备高算力、高能效、高通用性等优势,训推一体,已在多个行业完成大规模商业化落地,其中在中国电信落地了国内少有的真正实现市场化建设和运营的国产千卡集群,连续训练30天不中断,断点续训小于5分钟,支持客户完成大模型训练,稳定性、性能、精度各项指标达到客户要求。此次基于壁砺TM106系列产品,壁仞科技快速高效完成了DeepSeek-V3各项关键技术落地,基于RedPajama-Data-1T-Sample主流开源数据集实现了端到端高效预训练,Loss收敛正常。
07.实现W4A16细粒度量化技术,支持
DeepSeek-V3/R1满血版高效推理
壁仞科技在此前已经支持了BF16、INT8等版本的DeepSeek-V3/R1满血版推理,此次新提出W4A16等细粒度混合量化技术,进一步降低了处理延时,提升了总吞吐。另外针对DeepSeek官方最新发布的推理系统参考架构,壁仞科技已初步实现了分布式EP以及PD分离架构,将进一步提供更高性价比的推理集群解决方案。
08.One More Thing
业界首创大模型弹性训推一体架构,
实现集群最佳利用效率
DeepSeek 开放周最后一天公布了One More Thing-DeepSeek-V3/R1推理系统参考方案,为行业展示了如何实现高性能、低成本的分布式推理集群。其提到“由于白天的服务负荷高,晚上的服务负荷低,因此我们实现了一套机制,在白天负荷高的时候,用所有节点部署推理服务。晚上负荷低的时候,减少推理节点,以用来做研究和训练”,在计算推理成本时,其将释放给训练使用的资源排除在外,因此降低了推理成本(大概20%= 100% - 226节点/278节点)。但如何实现训练和推理任务高效共享资源呢?
此前在落地千卡集群时,壁仞科技已业界首创了大模型3D并行训练任务自动弹性伸缩机制,可以在大模型训练任务不中断的情况下,根据集群的资源情况自动进行在线扩缩容。根据业务流量情况自动进行削峰填谷,对大模型推理服务进行实例缩容、扩容,相应的训练任务则自动进行扩容、缩容。通过这套弹性训推一体架构,可以将集群资源的使用率保持在近100%,大幅降低资源空闲浪费成本。
壁仞科技打造了软硬一体、全栈优化、异构协同、开源开放的大规模智算集群全栈解决方案,凭借其技术创新能力和商业化落地成果,先后获得环球时报新质生产力产业实践“人工智能”示范标杆等多个权威技术奖项,技术方案和产品已在运营商、智算中心、行业AI、金融、电力等多个行业规模落地。此次通过八大核心技术赋能DeepSeek-V3满血版全栈式训练推理,进一步提升壁仞科技软硬一体、开箱即用的产品实力,支持客户利用好DeepSeek的新质生产力实现业务创新落地!
(责任编辑:热点)
-
量子计算具有变革药物发现、物流等各行各业的巨大潜力。然而,噪声一直是量子器件在投入实际应用的过程中所面临的巨大障碍。由于存在这些因环境相互作用和不完美硬件导致的噪声干扰,量子计算的性能目前在量子比特仅
...[详细]
-
开栏的话:今年是“十五五”开局之年。为营造良好舆论氛围,本报今起推出“奋进‘十五五’·滁州新篇章”专栏,聚集建设&
...[详细]
-
4月8日,湖南三岳数维科技有限公司以下简称“三岳数维”)向西北工业大学机电学院的捐赠仪式在西北工业大学友谊校区举行。此次捐赠核心为三岳数维自主研发的i3D Act实时三维引擎及
...[详细]
-
滁州网讯全媒体记者邢 颖 见习记者陈志璇)6月12日,市发改委联合人民银行滁州市分行、琅琊区发改委举办以“珍视信用价值,争创信用滁州”为主题的“6·1
...[详细]
-
回顾2024年可穿戴芯片新品:AI技术迭代加速,医疗监测成为发展新主线
电子发烧友网报道文/莫婷婷)2024年,可穿戴设备市场持续发展,上游各家厂商创新不断,在芯片解决方案、传感器技术、无线通信模块等方面不断创新,以满足日益增长的功能需求,例如更精准的健康监测、更长的电池
...[详细]
-
“‘引江入滁’供水保障工程全线贯通,将彻底改写滁州‘水困’历史,为滁州人民带来福祉,是树立和践行正确政绩观的生动实践。”6月10
...[详细]
-
滁州网讯全媒体记者喻 松 通讯员付 欣)近日,市退役军人事务局紧扣消费与就业需求,接连举办三场退役军人“新农人”军创产品展销会暨专场招聘会系列活动。以跨区域联动、产销对接、岗位
...[详细]
-
明光市坚持以正确政绩观答好民生答卷——把民生清单转化为幸福账单“老电梯用了十几年,现在换了新电梯又稳又安静,下楼散步、买菜都安心多了!”站在崭新的电梯里
...[详细]
-
这个被称为 QUEEN 的模型支持低带宽、高质量的场景生成,可用于工业机器人操作、3D 视频会议和直播等流媒体应用。NVIDIA Research 与马里兰大学合作开发的 AI模型 QUEEN 将内容
...[详细]
-
走进枞阳县经开区生产一线,机器轰鸣、布料翻飞间,疏红桃步履稳健、穿梭忙碌,时而查看生产流程,时而与一线员工交流叮嘱。作为安徽玉堂新材料科技有限公司党支部书记、总经理,疏红桃始终以党员初心、企业家担当、
...[详细]

乌衣镇多措并举提升物业服务水平_
民族音乐剧《再唱洗衣歌》在西藏大剧院震撼上演
南谯“直播+”创新基层治理模式_
凤阳“三箭齐发”推动产业优化_
燧原科技亮相2024中国人工智能大会