安卓 GPU 加速进阶:如何通过自定义 OpenGL ES Shader 实现推理库不支持的核心算子
前言 在安卓端侧推理(如使用 MNN、NCNN 或 TFLite)时,经常会遇到框架不支持某些特殊算子(如特定的激活函数、自定义的特征融合层)的情况。此时,如果回退到 CPU 执行会造成严重的性能瓶颈。本文将介绍如何编写一个 OpenGL ...
前言 在安卓端侧推理(如使用 MNN、NCNN 或 TFLite)时,经常会遇到框架不支持某些特殊算子(如特定的激活函数、自定义的特征融合层)的情况。此时,如果回退到 CPU 执行会造成严重的性能瓶颈。本文将介绍如何编写一个 OpenGL ...

引言:大模型推理为何成为部署瓶颈 2024年以来,大语言模型(LLM)的能力持续攀升,从GPT-4、Claude 3到Llama 3、Qwen2,模型参数量动辄数百亿,推理成本居高不下。当企业从”调API体验”转向&...
如何解决国产 AI 芯片下的集合通信瓶颈:深度对比 HCCL 与 NCCL 在 Ring AllReduce 上的差异 在分布式深度学习中,集合通信(Collective Communication)是决定训练效率的关键。在 NVIDIA ...
深入理解 Android NNAPI 的中间层调度机制 Android Neural Networks API (NNAPI) 是 Google 为 Android 设备提供的一套用于运行计算密集型机器学习模型的框架。它的核心价值在于提供了...
如何在海光 DCU DTK 环境下平滑迁移 ROCm 项目:解决版本冲突的实战指南 海光 DCU(Deep Computing Unit)作为国产 AI 算力的佼佼者,其底层软件栈 DTK(Deep-learning ToolKit)高度兼...
如何实现 AI 模型权重的端到端加密部署:从服务器下发到 TEE 内部解密的完整链路 在 AI 模型商业化落地中,模型权重是核心资产。如果模型直接以 .onnx 或 .tflite 明文形式存储在终端磁盘,极易被破解者直接“拖库”。本文将介...
如何针对大模型特定层进行混合精度量化:基于敏感度分析的自动策略分配 在部署大语言模型(LLM)时,全量化(如统一 INT4)虽然能极大降低显存占用,但往往会导致模型在复杂逻辑推理上出现“降智”。由于模型不同层对精度的敏感度不同,混合精度量化...
在使用 TensorFlow (TF) 进行深度学习开发时,尤其是涉及到 GPU 资源管理时,许多开发者会遇到一个令人头疼的问题:TensorFlow 默认会在初始化时,预先分配几乎所有可用的 GPU 显存,即使模型非常小。这导致了显存资源...
如何使用 CANN 插件在国产昇腾 NPU 上快速迁移并加速 PyTorch 模型 随着国产化算力需求的爆发,将现有的 AI 模型从 CUDA 环境迁移到国产昇腾(Ascend)平台已成为许多开发者的核心任务。得益于华为提供的 CANN(C...
1. 背景:移动端异构架构的挑战 在移动端 SoC 中,常见的 ARM 架构通常采用 Big.LITTLE(大小核)设计。当运行 AI 模型推理时,如果系统将计算任务随机分配给小核,或者在大小核之间频繁切换,会导致推理耗时出现显著的“长尾效...