0

0

DeepSeek开源FlashMLA,从GPU架构进行优化开始......

星夢妙者

星夢妙者

发布时间:2025-04-23 18:54:01

|

834人浏览过

|

来源于php中文网

原创

deepseek终于推出了他们的首个开源项目,flashmla,这是一个针对nvidia hopper架构gpu优化的高效多头线性注意力(mla)解码内核,专门为处理可变长度序列设计。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......中国版的OpenAI——DeepSeek,在全球引发热议后,发布了他们的首个开源项目。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashMLA是DeepSeek针对NVIDIA Hopper架构GPU优化的高效MLA解码内核,专为处理可变长度序列而设计。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......通过优化KV缓存机制和使用BF16数据格式,FlashMLA提升了内存和计算效率。在H800 SXM5 GPU上,其内存带宽可达3000 GB/s,计算性能可达580 TFLOPS。

FlashMLA项目正是幻方AI应用工程师之前提到的萤火AI平台优化英伟达算力集群的技术,现在已经开源!

FlashMLA的设计灵感来源于FlashAttention 2&3和Cutlass项目,支持分页缓存和低秩压缩等技术,进一步优化了内存管理和计算性能。

它结合了FlashAttention的高效内存管理和并行化策略,以及CUTLASS的底层优化和架构支持,从而在Hopper架构上实现了高效的多头注意力解码。这种结合使得FlashMLA在变长序列场景下表现出色,同时充分利用了现代GPU的硬件特性。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashAttention是具有IO感知功能的快速且节省内存的Exact Attention。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashAttention-3针对Hopper GPU(例如H100)进行了优化,并进行了相关测试。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashAttention-3在H100 80GB SXM5上使用FP16时速度提升。FlashMLA借鉴了FlashAttention的分页缓存机制,优化了缓存的管理和访问效率,特别是在变长序列场景下。

FlashAttention-2和FlashAttention-3的并行化和工作分配策略为FlashMLA提供了优化思路,使其能够在Hopper架构上实现高效的计算。

FlashAttention对BF16和FP16的支持也为FlashMLA提供了参考,使其能够充分利用现代GPU的硬件特性。

CUTLASS是由NVIDIA开发的高性能CUDA模板库,用于实现矩阵乘法(GEMM)和相关计算。

提供了高度可定制的模板抽象,支持多种数据类型(如FP64、FP32、FP16、BF16等)和混合精度计算。

MvMmall 网店系统
MvMmall 网店系统

免费的开源程序长期以来,为中国的网上交易提供免费开源的网上商店系统一直是我们的初衷和努力奋斗的目标,希望大家一起把MvMmall网上商店系统的免费开源进行到底。2高效的执行效率由资深的开发团队设计,从系统架构,数据库优化,配以通过W3C验证的面页模板,全面提升页面显示速度和提高程序负载能力。3灵活的模板系统MvMmall网店系统程序代码与网页界面分离,灵活的模板方案,完全自定义模板,官方提供免费模

下载

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashMLA的部分优化灵感来源于CUTLASS的特性:

底层优化:CUTLASS提供了高效的矩阵乘法内核和底层优化策略,FlashMLA借鉴了这些优化思路,以实现高效的多头注意力计算。架构支持:CUTLASS对多种NVIDIA GPU架构的支持为FlashMLA提供了硬件适配的基础,使其能够针对Hopper架构进行优化。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......模板化设计:CUTLASS的模板化设计思想为FlashMLA提供了代码结构和模块化的参考,使其能够灵活地实现不同的优化策略。适用于大语言模型(LLM)的推理任务,在需要高效解码的自然语言处理(NLP)场景中表现出色。开发者可以通过简单的安装命令快速部署,运行基准测试脚本验证性能。

DeepSeek开源FlashMLA,从GPU架构进行优化开始......FlashMLA需要以下硬件和软件环境:

硬件:NVIDIA Hopper架构GPU(例如H800 SXM5)。软件:CUDA 12.3及以上版本;PyTorch 2.0及以上版本。安装FlashMLA:通过以下命令安装FlashMLA:

python setup.py install

运行基准测试脚本:

python tests/test_flash_mla.py

使用CUDA 12.6,在H800 SXM5上实现高达3000 GB/s的内存绑定配置和580 TFLOPS的计算绑定配置。

看看DeepSeek开源的项目,这才是真正的OpenAI!

DeepSeek正在以实际的行动践行"开源、AI先锋"的使命,这是国运级别的创新,也是本土诞生的幸事。

"若前方无路,我便踏出一条路,若天理不容,我便逆转这乾坤",

从DeepSeek爆火的背后,看到的是哪吒的不屈精神,誓要打破这层技术壁垒,为后来者镗出一条路来。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
数据类型有哪几种
数据类型有哪几种

数据类型有整型、浮点型、字符型、字符串型、布尔型、数组、结构体和枚举等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

309

2023.10.31

php数据类型
php数据类型

本专题整合了php数据类型相关内容,阅读专题下面的文章了解更多详细内容。

222

2025.10.31

pytorch是干嘛的
pytorch是干嘛的

pytorch是一个基于python的深度学习框架,提供以下主要功能:动态图计算,提供灵活性。强大的张量操作,实现高效处理。自动微分,简化梯度计算。预构建的神经网络模块,简化模型构建。各种优化器,用于性能优化。想了解更多pytorch的相关内容,可以阅读本专题下面的文章。

432

2024.05.29

Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

24

2025.12.22

Python 自然语言处理(NLP)基础与实战
Python 自然语言处理(NLP)基础与实战

本专题系统讲解 Python 在自然语言处理(NLP)领域的基础方法与实战应用,涵盖文本预处理(分词、去停用词)、词性标注、命名实体识别、关键词提取、情感分析,以及常用 NLP 库(NLTK、spaCy)的核心用法。通过真实文本案例,帮助学习者掌握 使用 Python 进行文本分析与语言数据处理的完整流程,适用于内容分析、舆情监测与智能文本应用场景。

23

2026.01.27

php中文乱码如何解决
php中文乱码如何解决

本文整理了php中文乱码如何解决及解决方法,阅读节专题下面的文章了解更多详细内容。

1

2026.01.28

Java 消息队列与异步架构实战
Java 消息队列与异步架构实战

本专题系统讲解 Java 在消息队列与异步系统架构中的核心应用,涵盖消息队列基本原理、Kafka 与 RabbitMQ 的使用场景对比、生产者与消费者模型、消息可靠性与顺序性保障、重复消费与幂等处理,以及在高并发系统中的异步解耦设计。通过实战案例,帮助学习者掌握 使用 Java 构建高吞吐、高可靠异步消息系统的完整思路。

1

2026.01.28

Python 自然语言处理(NLP)基础与实战
Python 自然语言处理(NLP)基础与实战

本专题系统讲解 Python 在自然语言处理(NLP)领域的基础方法与实战应用,涵盖文本预处理(分词、去停用词)、词性标注、命名实体识别、关键词提取、情感分析,以及常用 NLP 库(NLTK、spaCy)的核心用法。通过真实文本案例,帮助学习者掌握 使用 Python 进行文本分析与语言数据处理的完整流程,适用于内容分析、舆情监测与智能文本应用场景。

23

2026.01.27

拼多多赚钱的5种方法 拼多多赚钱的5种方法
拼多多赚钱的5种方法 拼多多赚钱的5种方法

在拼多多上赚钱主要可以通过无货源模式一件代发、精细化运营特色店铺、参与官方高流量活动、利用拼团机制社交裂变,以及成为多多进宝推广员这5种方法实现。核心策略在于通过低成本、高效率的供应链管理与营销,利用平台社交电商红利实现盈利。

120

2026.01.26

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.3万人学习

Django 教程
Django 教程

共28课时 | 3.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号