
PyTorch 2.0 torch.compile 深度解析:从原理到生产部署的完整指南
引言:为什么 torch.compile 改变了 PyTorch 的游戏规则 2023 年 PyTorch 2.0 的发布标志着 PyTorch 生态的一个重大转折点。其中最核心的新特性—— 1torch.compile ——被 PyTor...

引言:为什么 torch.compile 改变了 PyTorch 的游戏规则 2023 年 PyTorch 2.0 的发布标志着 PyTorch 生态的一个重大转折点。其中最核心的新特性—— 1torch.compile ——被 PyTor...
如何利用 ARM i8mm 指令集加速端侧量化模型推理 在端侧 AI 推理(如手机、嵌入式设备)中,Int8 量化是提升性能、降低能耗的核心技术。传统的 ARM NEON 指令集虽有 SDOT (点乘) 指令,但在处理大规模矩阵乘法时仍显吃...
如何通过 AclLite 封装大幅简化昇腾推理程序的开发流程 在国产昇腾(Ascend)芯片上进行 AI 推理开发时,开发者通常需要直接面对 CANN (Compute Architecture for Neural Networks) 的...
如何针对瑞芯微 RK3588 NPU 进行算子裁剪与加速:实现边缘 AI 的极致响应 在边缘计算领域,瑞芯微 RK3588 以其 6TOPS 的 NPU 算力成为国产芯片的佼佼者。然而,许多开发者发现直接部署模型时,推理速度远达不到预期。这...
如何利用华为迁移工具自动化实现 CUDA 到 CANN 的算子代码映射 随着国产算力加速卡的应用普及,将原本运行在 NVIDIA GPU 上的 CUDA 算子迁移到华为昇腾 Ascend NPU 环境已成为开发者的核心痛点。本文将详解如何利...
如何通过 ncnn 算子融合黑魔法减少推理时的内存读写 在端侧推理优化中,算力往往不是唯一的瓶颈,内存带宽(Memory Bandwidth)才是。在 ncnn 推理框架中,虽然 ncnnoptimize 工具已经能自动处理大部分算子融合(...
在移动端 AI 推理领域,MNN(阿里巴巴)和 NCNN(腾讯)是两个最受欢迎的深度学习推理框架。它们的性能差异往往取决于底层的硬件加速能力和驱动适配情况,尤其是在面对高通(Qualcomm Adreno)和联发科(MediaTek Mal...

在深度学习模型规模不断膨胀的今天,训练一个大型模型动辄需要数十甚至上百 GPU·天。当显存成为瓶颈、训练时间成为成本大头时,混合精度训练(Mixed Precision Training)就成了必须掌握的优化手段。它可以在几乎不损失模型精度...

一、引言:从 RLHF 到 DPO 的范式转变 大语言模型(LLM)在预训练阶段通过海量文本学习到了丰富的语言知识和世界知识,但预训练模型的行为并不一定符合人类期望——它可能输出有害内容、编造事实,或者无法遵循指令。为了让模型”...
如何解决昇腾 NPU 上频繁创建张量导致的内存碎片问题 在将模型从 CUDA 迁移到昇腾 NPU(Ascend)时,很多开发者会遇到一个诡异现象:通过 nvidia-smi 类似的工具观察,显存(HBM)占用并没满,但程序却频繁报出 Out...