云天励飞DeepEdge10适配DeepSeek开源周“首个大招”:FlashMLA
2026-03-16
share to :

2月24日,DeepSeek"开源周"正式启动,并发布首个代码库FlashMLA,引发全球关注。


FlashMLA是针对英伟达Hopper GPU优化的高效MLA解码内核,专为可变长度序列优化设计,是DeepSeek高效使用算力的核心程序模块。FlashMLA通过MLA解码优化与分页KV缓存技术,显著提升硬件利用率,加速大语言模型解码过程,有效提高响应速度与吞吐量,适用于各种实时对话生成场景。


云天励飞芯片团队在FlashMLA开源后,迅速完成了DeepEdge10平台与FlashMLA的适配工作。在适配过程中,云天励飞采用了Op fusion tiling、Online softmax、Double buffer、细粒度存算并行等先进技术,并基于自研的Triton-like编程语言快速开发验证了高效的FlashMLA算子。通过一系列优化,不仅显著提升了计算效率,还大幅降低了显存占用,充分展现了DeepEdge10平台“算力积木”芯片架构的卓越优势,以及其与DeepSeek生态的高度契合性。


今日,云天励飞已将相关代码提交至开源平台Gitee,为开源AI贡献了重要的技术力量。


算子的源码地址为:

https://gitee.com/Intellifusion_2025/tyllm/blob/master/python/tylang/flash_mla.py


DeepEdge10系列芯片是专门针对大模型时代打造的芯片,支持包括 Transformer 模型、BEV 模型、CV 大模型、LLM 大模型等各类不同架构的主流模型;基于自主可控的先进国产工艺打造,采用独特的“算力积木”架构,可灵活满足智慧城市、智慧交通、智能制造、智慧仓储、机器人、边缘智算中心等不同场景对算力的需求,为大模型推理提供强大动力。


1.jpg


目前,DeepEdge10芯片平台已成功适配DeepSeek R1系列模型及国产鸿蒙操作系统。未来,云天励飞将持续加大研发力度,推动芯片在国产化领域取得更大突破,为国产AI生态建设贡献更多力量。


深圳总部
深圳市南山区粤海街道深圳湾科技生态园10栋B座14-15楼、33楼
青岛分部
青岛市崂山区科苑纬一路1号国际创新园D2座2601室
成都天府励飞科技有限公司
成都市双流区西航港街道临港路一段32号成都东航中心T2栋808室
上海分部
上海市浦东新区张江高科技园区川和路55弄人工智能岛21栋301室
南京分部
南京市江北新区星火路创智大厦A座301室