logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

昇腾Ascend C高阶实战:实现带Mask的Attention Score算子(支持大模型推理场景)

输入说明query,FP16key,FP16mask或(广播支持),FP16输出说明score,FP16✅关键特性支持动态序列长度(S_q, S_kv 可变)支持广播式 Mask(节省显存)自动处理Scale 缩放1D1/\sqrt{D}1/D​本文通过实现Fused Attention Score 算子✅ 如何用 Ascend C 解决大模型推理中的真实性能瓶颈✅分块计算(Tiling)与在线融

文章图片
#c语言#开发语言
FlashAttention与Gradient Checkpointing:显存不够时怎么训练超大模型?

"""细粒度Gradient Checkpointing策略:不是整个Attention作为一个检查点而是按block分层,每B个block保存一个检查点好处:反向时只需重计算B个block,不是整个序列"""self.checkpoint_every = checkpoint_every # 每N个block保存一个检查点"""细粒度检查点前向把序列分成多个block,每checkpoint_e

文章图片
#python#深度学习#机器学习
FlashAttention与Gradient Checkpointing:显存不够时怎么训练超大模型?

"""细粒度Gradient Checkpointing策略:不是整个Attention作为一个检查点而是按block分层,每B个block保存一个检查点好处:反向时只需重计算B个block,不是整个序列"""self.checkpoint_every = checkpoint_every # 每N个block保存一个检查点"""细粒度检查点前向把序列分成多个block,每checkpoint_e

文章图片
#python#深度学习#机器学习
FlashAttention多模态应用:图像token太多时Attention怎么算?

某团队在昇腾NPU上跑多模态大模型(LLaVA、Qwen-VL等),发现FlashAttention的效果没有纯文本场景那么明显。他们分析了一下原因:多模态模型的文本部分不长(几百个token),但图像部分被tokenize之后有几百到几千个token,这些token之间的关系很复杂——局部token之间关系紧密(如同一图像patch内的token),远距离token之间关系稀疏(如不同图像pat

文章图片
#计算机视觉#人工智能#深度学习
FlashAttention与AWQ量化:模型权重怎么压缩又不掉精度?

某团队在昇腾NPU上部署量化后的模型(INT8),发现模型质量下降明显——生成文本的流畅度不如FP16,数学计算的正确率也下降了。他们试过GPTQ量化,效果稍好但仍然有明显损失。问题出在量化的策略上。GPTQ和标准INT8量化都只看权重的分布,没有考虑激活值(activation)的分布。权重分布均匀,但激活值的分布差异很大——某些权重虽然数值小,但对应的激活值很大,这些权重如果被粗量化,会严重影

文章图片
#人工智能#python
FlashAttention与AWQ量化:模型权重怎么压缩又不掉精度?

某团队在昇腾NPU上部署量化后的模型(INT8),发现模型质量下降明显——生成文本的流畅度不如FP16,数学计算的正确率也下降了。他们试过GPTQ量化,效果稍好但仍然有明显损失。问题出在量化的策略上。GPTQ和标准INT8量化都只看权重的分布,没有考虑激活值(activation)的分布。权重分布均匀,但激活值的分布差异很大——某些权重虽然数值小,但对应的激活值很大,这些权重如果被粗量化,会严重影

文章图片
#人工智能#python
FlashAttention与AWQ量化:模型权重怎么压缩又不掉精度?

某团队在昇腾NPU上部署量化后的模型(INT8),发现模型质量下降明显——生成文本的流畅度不如FP16,数学计算的正确率也下降了。他们试过GPTQ量化,效果稍好但仍然有明显损失。问题出在量化的策略上。GPTQ和标准INT8量化都只看权重的分布,没有考虑激活值(activation)的分布。权重分布均匀,但激活值的分布差异很大——某些权重虽然数值小,但对应的激活值很大,这些权重如果被粗量化,会严重影

文章图片
#人工智能#python
FlashAttention与AWQ量化:模型权重怎么压缩又不掉精度?

某团队在昇腾NPU上部署量化后的模型(INT8),发现模型质量下降明显——生成文本的流畅度不如FP16,数学计算的正确率也下降了。他们试过GPTQ量化,效果稍好但仍然有明显损失。问题出在量化的策略上。GPTQ和标准INT8量化都只看权重的分布,没有考虑激活值(activation)的分布。权重分布均匀,但激活值的分布差异很大——某些权重虽然数值小,但对应的激活值很大,这些权重如果被粗量化,会严重影

文章图片
#人工智能#python
FlashAttention与多模态:图文联合理解实战

FlashAttention通过跨模态Attention、模态融合、图文Embedding映射,让多模态理解的显存降低65%,推理速度提升4.2倍,图文检索准确率提升11.8%。在昇腾NPU上,还有Cube/Vector并行、达芬奇架构感知模态融合、零拷贝跨模态数据传输等独有优化。如果你在做多模态理解(比如图文问答、图文检索、视觉常识推理),需要同时处理文本和图片,试试多模态FlashAttent

文章图片
#人工智能#spring#后端
FlashAttention与多模态:图文联合理解实战

FlashAttention通过跨模态Attention、模态融合、图文Embedding映射,让多模态理解的显存降低65%,推理速度提升4.2倍,图文检索准确率提升11.8%。在昇腾NPU上,还有Cube/Vector并行、达芬奇架构感知模态融合、零拷贝跨模态数据传输等独有优化。如果你在做多模态理解(比如图文问答、图文检索、视觉常识推理),需要同时处理文本和图片,试试多模态FlashAttent

文章图片
#人工智能#spring#后端
    共 49 条
  • 1
  • 2
  • 3
  • 4
  • 5
  • 请选择