已关闭
[Requirement|需求建议]: 新增NPU算子执行链路拆解文档 #166
ren-botao创建于  5月15日关闭于  5月18日
ren-botao
5月15日 创建

一、背景信息 (必填)

新增一份详细的技术文档,描述 NPU 算子从 PyTorch 调用到芯片执行的完整链路。该文档位于 Samples/0_Introduction/npu_execution/README.md,涵盖三个主要阶段:

  1. PyTorch 框架调用阶段分析:包括初始化、THPVariable_xxx流程、Dispatch流程、Redispatch流程、torch_npu侧执行流程
  2. ACLNN单算子、NNOPBASE框架调用阶段分析:包括ACLNN二段式接口、INFER_SHAPE机制、Tiling机制、Workspace计算、Kernel下发执行
  3. AICore算子下发阶段分析:包括算子调度部分、算子执行部分

二、价值/作用 (必填)

帮助开发者深入理解 CANN 软件栈中算子的完整执行流程,包括:

  • PyTorch 端的 Dispatch 机制与 torch_npu 的算子注册
  • ACLNN 二段式接口的工作原理
  • INFER_SHAPE、Tiling、Workspace 等关键机制
  • AICore 硬件执行流程

为算子开发、性能优化、问题排查提供底层原理参考。

三、设计方案 (必填)

3.1 使能方式

本文档为纯技术文档,不涉及算子使能方式。

3.2 总体设计
3.2.1 算子支持的数据类型

不适用(文档类型)

3.2.2 host侧设计

不适用(文档类型)

3.2.3 kernel侧设计

不适用(文档类型)

3.3 支持硬件

文档内容适用于华为昇腾 NPU 平台(AICore)

3.4 算子约束限制

不适用(文档类型)

💡 备注(选填)

likedislike
Rren-botao
5月15日 添加了label:requirement
Rren-botao
5月15日 修改了issue 的描述
ren-botao
5月15日 评论:

/assign

likedislike
CANN-robotCANN-robot成员
5月15日 将 ren-botao 设为负责人
CANN-robotCANN-robot成员
5月18日 关闭了 issue
CANN-robotCANN-robot成员
5月18日 添加了label:resolved