首先完成在八款芯片适配的DeepSeek-V4-Flash 是深度求索推出的 V4 系列两大模型之一,采用混合专家(MoE)架构,总参数量 284B,激活参数仅 13B,支持 100 万 token上下文长度。该模型在架构上引入了混合注意力机制(结合压缩稀疏注意 力 CSA 与高度压缩注意力HCA,大幅提升长上下文效率)、流形约束超连接(mHC,增强跨层 信号传播稳定性)以及 Muon优化器(加速收敛、提升训练稳定性)。预训练数据超过 32T token,后训练采用两阶段范式——先通过 SFT和 GRPO 强化学习独立培养领域专家,再通过在线策略蒸馏将多领域能力统一整合到单一模型中。在最大推理力度模式(Flash-Max)下,给予更大思考预算使其推理能力可接近 Pro版本水平;受限于参数规模,在纯知识类任务和最复杂的 Agent 工作流上略逊于 Pro。 整体性能参考如下官方评测结果:
围绕 DeepSeek-V4-Flash多芯适配,此次 FlagOS 系统软件技术栈突破了三大关键技术:FlagGems 全算子替代(实现多芯片统一适配)、为o-group采用独立张量并行策略解锁更多低显存场景、以及 “FP4+FP8混合精度”的原生权重到 FP8/BF16 的精度路径转换。当下国内出货的AI芯片,都没有FP4的支持。英伟达也只有在Blackwell及之后的高端芯片才支持FP4。这三项关键技术,使得 DeepSeek V4 能够在当前各种厂商的主流 AI 芯片上稳定运行,而非仅限于支持 FP4 和大显存的少数高端AI加速卡。
一、开发者速用指南:DeepSeek-V4-Flash 多芯版本获取与部署
FlagOS技术栈为 DeepSeek-V4-Flash 新模型提供了用户提供开箱即用的多芯片 DeepSeek-V4-Flash 版本。目前FlagOS的统一算子库FlagGems、统一编译器FlagTree及基于FlagScale的多芯片适配支持下,海光、沐曦、华为、摩尔线程(FP8)、昆仑芯、平头哥、天数、英伟达(FP8)等八款芯片,已经完成 DeepSeek-V4-Flash的跨芯适配及验证。使用源码进行安装部署,可参考以下指引:
GitHub:https://github.com/flagos-ai/DeepSeek-V4-FlagOS
方式一:FlagOS安装部署
快速安装
安装FlagOS算子库
官方仓库:https://github.com/flagos-ai/FlagGems
pip install flag-gems==5.0.2安装FlagOS编译器
官方仓库:https://github.com/flagos-ai/flagtree
# 安装命令以英伟达平台为例:python3 -m pip uninstall -y tritonpython3 -m pip install flagtree===0.5.0 --index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple
推理部署
单机(8卡)
export USE_FLAGGEMS=1 #开启加速torchrun --nproc-per-node 8 generate.py --max-new-tokens 64 --ckpt-path /path/to/model_bf16_mp8 --config config_from_bf16.json --input-file prompt.txt
双机(16卡)
Node 0:
export NCCL_SOCKET_IFNAME=eth0export NCCL_IB_DISABLE=1export USE_FLAGGEMS=1torchrun --nnodes=2 --nproc_per_node=8 --node_rank=0 --master_addr=<master_ip> --master_port=29500 generate.py --ckpt-path /path/to/model_bf16_mp16 --config config_from_bf16.json --input-file prompt.txt --max-new-tokens 64
Node 1:
export NCCL_SOCKET_IFNAME=eth0export NCCL_IB_DISABLE=1export USE_FLAGGEMS=1torchrun --nnodes=2 --nproc_per_node=8 --node_rank=1 --master_addr=<master_ip> --master_port=29500 generate.py --ckpt-path /path/to/model_bf16_mp16 --config config_from_bf16.json --input-file prompt.txt --max-new-tokens 64
方式二:模型镜像直接下载
用户可以直接拉取在 FlagRelease 上发布的迁移后的模型文件、代码和镜像。以下是迁移适配后的几种 AI 芯片的模型版本,开箱即用、无需迁移。
魔搭平台
HuggingFace 平台
二、三大技术突破:为什么对支持多种AI芯片十分重要
突破一:FlagGems 提供支持8种以上芯片的全算子替代——真正意义上的跨芯方案
本次 DeepSeek-V4-Flash 的适配,FlagGems 实现了模型推理链路中全部算子的替代。这意味着什么?
彻底脱离 CUDA 算子依赖:DeepSeek-V4-Flash的 MoE 专家调度、Attention 计算、RMSNorm、TopK 路由等全部核心计算模块,均由 FlagGems 基于 Triton/Triton-TLE语言重新实现,不调用任何 cuDNN/cuBLAS 等 NVIDIA 私有库。
无需芯片厂商逐一适配:传统模式下,每款新模型上线,芯片厂商需要投入工程团队做算子适配。现在通过 FlagGems + FlagTree 编译器的组合,新模型的算子可以直接编译到多款芯片后端,芯片厂商不需要做任何额外工作。
新算子即时可用:DeepSeek-V4-Flash引入的新计算模式(如 o-group 相关的分组路由机制),FlagGems 已经实现了对应的新算子,并通过 FlagTree 编译器统一编译到所有支持的芯片后端。
FlagGems 作为全球最大的 Triton 单一算子库,已拥有超过400 个大模型常用算子,并已正式进入 PyTorch 基金会生态合作项目。在 40 个主流模型上,推理任务算子覆盖度达到 90%~100%,完整支持 DeepSeek-V4-Flash的全部计算需求。
突破二:为o-group采用独立并行策略——解除张量并行最多单机8卡限制
DeepSeek-V4-Flash为了进一步降低计算开销采用了分组输出投影技术(Grouped Output Projection),配置为o_group=8,这导致在传统的张量并行时候,最多切8份。而当前一些主流国产芯片的单卡显存为 32GB 或 64GB,尤其在BF16格式情况下,需要张量并行大于8份才能放的下。为了解除这个限制,FlagOS专门针对o_groups进行了单独张量并行策略设计和实现,确保o_goups切分不超过8份的前提下,能够让模型其他部分还采用经典的张量并行策略,并且实现超过8份的切分。通过不同的张量并行策略组合,能够实现多于8台设备的张量并行运行。
FlagOS 团队对o_group张量并行改动有:
独立的并行策略:独立于已有的张量并行通信组之外,为o-group单独构建所需要的张量并行通信组,确保其他模型结构张量并行切分超过8的情况下,o-group的张量并行在8以内。
参数转换调整:对o_group相关的参数,也进行了对应单独的张量并行切分处理,以确保在新的独立张量并行策略下,也能够被正确加载。
覆盖面扩展:这一优化能够将 DeepSeek-V4-Flash在单独采用张量并行策略下,将可运行芯片范围从"仅限单机80GB以上显存的个别高端卡"扩展到"多机64GB/32GB的更多主流国产芯片",包括海光、沐曦、天数智芯等厂商的主力产品线。
突破三:从“FP4+FP8混合精度” 到 BF16的精度转换——打通主流芯片的计算路径
DeepSeek-V4-Flash模型发布时首次采用 FP4+FP8混合精度,该精度只有在Blackwell及之后的英伟达最新硬件上才有支持,但当前所有国内非英伟达 AI 芯片都未能支持,只有摩尔线程原生支持了FP8,其余依然以BF16为主。
FlagOS 完成了从 FP4 到 BF16 的完整精度转换:
权重反量化:将 FP4 量化权重转换为 BF16 格式。这不是简单的类型转换,而是需要根据 DeepSeek 的量化方案进行逆量化计算,确保数值精度。
计算路径重建:FP4 和 BF16 在底层计算上有本质差异——FP4 的动态范围更窄,累加精度、溢出处理策略均不同。FlagOS 对推理链路中的 GEMM、Attention、MoE 路由等关键计算节点逐一适配了 BF16 路径。
精度对齐验证:经过标准评测集验证,BF16 版本与 FP4 原生版本在核心能力指标上保持对齐,确保精度转换不引入业务层面的效果损失。
本次,FlagOS推出了FP8和BF16两种适配版本,让DeepSeek-V4-Flash不再是"只有最新 NVIDIA 卡才能跑"的模型,而是真正可以部署在 FP8 及 BF16 生态的主流国产芯片上。
三、FlagGems开源高性能新算子 全面支持 DeepSeek-V4-Flash
本次新发布的DeepSeek-V4-Flash共有大约67个算子,FlagGems已全量支持。新支持了Act Quant、hc_split_sinkhorn、FP8 MatMul、Sparse Attention、Hadamard Transform等5个新算子,实现了对DeepSeek-V4-Flash的全面支持,也为跨芯适配打下重要基础。
FlagGems高性能算子库的下载使用
Gitcode:https://gitcode.com/flagos-ai/FlagGems/tags/v5.0.2
Gitee: https://gitee.com/flagos-ai/FlagGems/tree/v5.0.2
Github:https://github.com/flagos-ai/FlagGems/tree/v5.0.2
FlagGems 支持 DeepSeek-V4-Flash 新算子的性能对比
为了支持更多AI芯片的使用,FlagOS对DeepSeek-V4-Flash中使用的新算子使用Triton语言进行重新实现,基于FlagTree统一编译器,性能全部超过原生性能。
基于C++ Wrapper技术,推理性能全面提升
C++ Wrapper技术是FlagOS技术社区专门为提升基于Triton语言的算子内核调用效率而打造的技术。目前已经支持了该技术的芯片包括华为昇腾、寒武纪、摩尔线程、平头哥真武、及英伟达等。使用了C++ Wrapper技术,在普通的Transformers框架下,可以显著提升使用了Triton算子的模型的端到端效率,实现跨芯普适、和高效推理的双重目标。通过端到端效果评测(NV H20,DeepSeek-V4-Flash FP8),C++ Wrapper + Triton 比 TileLang 快 11%,比 Python Wrapper 版快 39%。
四、开发者极致体验:"发布即多芯" + "极简部署"
1. 核心能力与原生版本对齐
经 GPQA_Diamond、AIME等权威评测集验证,FlagOS 适配后的 DeepSeek-V4-Flash,在语言理解、复杂推理、代码生成、数学计算等核心能力上,与 CUDA 原生版本对齐,可放心应用于金融、教育、政企服务、代码开发等场景,无需担心适配导致业务效果折损。
评测数据:
注:本测试结果仅用于对迁移前(Nvidia-Origin)和迁移后(-FlagOS)版本的互相对齐验证,并不代表 DeepSeek 模型的官方性能,DeepSeek 模型的官方性能以 DeepSeek 官方公布数据为准。
2. 极简部署:开箱即用,底层优化无感知
FlagOS 将核心算子库、编译器等技术组件前置内置到 DeepSeek-V4-Flash代码框架中,开发者加载模型时,底层优化代码自动生效,无需手动添加任何 FlagOS 初始化代码。同时,基于 FlagRelease 直接提供了多芯片版本的 DeepSeek-V4-Flash-FlagOS 模型版本,标准化 Docker 镜像 + 一键加速命令,解决了开发者最头疼的环境配置、效果对齐、性能优化等问题。
五、FlagOS 2.0 技术底座:从大模型到智能体时代的全栈升级
DeepSeek-V4-Flash的三重突破,依托的是 FlagOS 2.0 统一多芯片系统软件栈的全链路能力。从算子层、编译层、框架层到工具层,全链路为大模型跨芯适配提供技术支撑,将原本数周的适配周期缩短至数天,真正实现极速落地。
FlagOS:面向多种 AI 芯片的系统软件栈
1. 高性能算子库 FlagGems:核心算子深度适配,释放硬件算力
FlagGems 作为 FlagOS 核心的高性能通用大模型算子库,基于 Triton 语言实现,针对 DeepSeek-V4-Flash推理链路的核心算子进行了深度适配与优化,包括 MoE 专家调度、Attention 计算、RMSNorm 等关键计算模块,同时原生支持 NVIDIA、摩尔线程、沐曦、清微智能、天数等接近 20 家 AI 芯片。
2. 统一 AI 编译器 FlagTree:一次编写,多芯编译
FlagTree 是 FlagOS 面向多 AI 芯片后端的统一编译器,基于 Triton 深度定制,可将 DeepSeek-V4-Flash的核心算子编译为英伟达、摩尔线程等十多种不同 AI 芯片后端可识别的指令,彻底解决不同芯片编译器生态割裂的问题,大幅降低算子跨芯片适配的开发成本。
六、开源共建:FlagOS持续做开发者的“跨芯适配后盾”
当下,"异构算力协同、大模型普惠落地"已成为全球开源开发者社区的核心热点,打破硬件生态隔离、让大模型在不同算力平台高效低成本运行,是无数开发者的核心诉求。FlagOS 从诞生之初就将开源开放、众智共建刻入技术基因,始终以开发者为中心,通过全栈开源的统一系统软件栈,把复杂的"M×N"硬件适配问题降维为"M+N",做每一位开发者最可靠的跨芯适配后盾。
全栈开源无保留,把技术主动权交给开发者
目前,FlagOS 已形成完整的开源技术体系,所有核心组件均已开源在 GitHub,同时开放了数十款最新的主流基础大模型、十多款 AI 芯片的适配方案与最佳实践,开发者可自由获取、深度定制:
多路径参与共建,全层级开发者均可入局
我们为不同技术方向、不同经验层级的开发者,设计了低门槛、多路径的共建方式,无论你是 AI 开发新手,还是深耕系统软件的资深专家,都能在 FlagOS 社区找到自己的位置。
关于众智 FlagOS 社区
为解决不同AI芯片大规模落地应用,北京智源研究院联合众多科研机构、芯片企业、系统厂商、算法和软件相关单位等国内外机构共同发起并创立了众智FlagOS社区,目前已经有78家成员单位。FlagOS是一款专为异构AI芯片打造的开源、统一系统软件栈,支持 AI 模型一次开发即可无缝移植至各类硬件平台,大幅降低迁移与适配成本。它包括大型算子库、统一AI编译器、并行训推框架、统一通信库等核心开源项目,致力于构建「模型-系统-芯片」三层贯通的开放技术生态,通过"一次开发跨芯迁移"释放硬件计算潜力,打破不同芯片软件栈之间生态隔离。
社区官网:https://flagos.io GitHub:https://github.com/flagos-ai GitCode:https://gitcode.com/flagos-ai SkillHub:https://skillhub.flagos.io

内容中包含的图片若涉及版权问题,请及时与我们联系删除









评论
沙发等你来抢