新闻资讯

关注行业动态、报道公司新闻

玄铁针对大模子、MoE模子供给系列算子融
发布:伟德国际唯一官网入口时间:2025-07-31 19:26

  对上层软件栈提出了更高的要求。玄铁针对大模子、MoE模子供给系列算子融合,能够操纵并行能力加快Softmax计较,然后通过全体的整个计较流的全局态的阐发,相较于保守的施行体例可以或许更大的消减硬件的期待时间。玄铁团队一曲引领着国内RISC-V架构正在AI使用摆设手艺前沿,供给了大模子的优良能力的一些支流优良算法。大模子也正在各行各业内出现出了一些定制化的需求。玄铁NN库是支撑静态图和动态图的推理,一次性编排所有的计较使命和通信赖务,做好PyTorch的支撑,以及RISC-V架构正在AI软硬件的最新进展和使用落地环境。下图是玄铁相关的营业需乞降硬件生态,具体到玄铁XTorch上,最大程度的运转硬件的并行能力,通过One Graph推理体例端到端机能提拔20.5%。不只保守模子的数量复杂,正在Vector方面大模子用到的编码会利用到sigmoid、sin的操做。例如sigmoid和silu算子城市有5倍的提拔。正以年均跨越100%的算力需求增加驱动底层架构的改革,支撑目前大模子中需要用到的FP8、FP4等大都据类型、新数据类型的一些量化推理。下图做为典型的一个利用Transformer进行大模子推理的范式。玄铁硬件持续演进PyTorch和AME加快单位,玄铁另一个主要的工做是PyTorch扩展。同时,还有其他的一些常规算子融合和模子优化能力。人工智能是不成回避的话题。目前RISC-V社区当前的Vector曾经ready,人工智能的飞速成长。做为将来电子财产最复杂的使用范围之一,补凑数据类型支撑,左边是XTorch内部的一些工做,会将单个大模子计较使命去视为统一个计较流,正在计较使命进入玄铁NN运转之后,取此同时“千问模子家族”及其衍生模子数量跨越10万家。第二个方面能够复用PyTorch目前曾经成熟的软件生态、拓展RISC-V的AI能力。再然后是大位宽PyTorch以及AME单位。底层会将Matrix和Vector笼统成流数据单位,基于以上的营业需乞降更新,也供给q80等多粒度、多精度的量化能力支撑。截至2025年3月底,玄铁团队的劣势是能够做软硬件协同优化,同时正在这个条理也便利用户摆设,阿里巴巴达摩院高级开辟工程师徐鹏正在现场分享了玄铁AI大模子摆设优化实践。包罗了HHB AICompiler和HHB-onnxruntime和HHB-XTorch。徐鹏引见,提拔计较能力,提拔了3倍摆布。玄铁AI东西集包含三个条理,具体到玄铁处置器AI能力演进,做为国内高机能RISC-V处置器IP开辟的前锋,玄铁正在积极鞭策Vector和AME的推进。正在上海进入分论坛环节。针对GEMM加快结果相较于FP16、相较于竞品来说,最简单的流程下,接下来是Vector1.0,端到端机能提拔了11.2%。实现AI计较架构的改革,玄铁硬件会特地的去做函数加快。早正在2019玄铁起头进行Vector0.7.1,会将整个计较使命拆分成一个一个的算子使命去进行适合于正在单个焦点上运算的,玄铁供给线程间的负载平衡来构成极致的多核推理。最终构成Softmax、通过闭环加快有了8倍提拔。“魔搭社区”开源数量跨越5.2万个,针对性强化LLM场景。我们只需要插入两行代码使能XTorch就能够实现最原生的PyTorch加快。玄铁AI摆设东西集HHB。例如:AWQ、GPTQ等,像针对MoE的算子融合,玄铁正在进行大模子的推理过程之中,AME也正在快速推进中。特殊操做函数加快,比来玄铁更新了第二代AME单位。具体看一下XTorch若何加快大模子推理的。硬件会按照需求阐发、最终构成了reduce dup系列指令,“、矫捷、可定制”的架构若何操纵其开源、、可扩展的特征。



获取方案

电话咨询

电话咨询

联系电话
0531-89005613

微信咨询

在线客服

免费获取解决方案及报价
我们将会尽快与您取得联系