OpPlugin of Ascend for PyTorch

文件	最后提交记录	最后更新时间
.gitcode	[docs]add PR notes info Co-authored-by: liangsongwei<liangsongwei@huawei.com> # message auto-generated for no-merge-commit merge: !4253 merge master into master [docs]add PR notes info Created-by: liangsongwei Commit-by: liangsongwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > (如有)请关联需求文档/issue链接 - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】一是要求必须关联社区issue链接；二是禁止填写需求链接、问题单等内部流程信息。该修改通过直接更新模板文件生效，不涉及任何代码逻辑变更。 # 【资料变更】 PR提交模板信息补充 # 【接口变更】不涉及 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7404318/6b276c27-f6bc-463b-a340-1c3e155aa1e3/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!4253	5 个月前
ci	test update ut Co-authored-by: huangyunlong2022<huangyunlong4@h-partners.com> # message auto-generated for no-merge-commit merge: !5181 merge cherry-pick-mr-5125-1781485906258-auto into 26.1.0 test update ut Created-by: huangyunlong2022 Commit-by: huangyunlong2022 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） im2col_backward fp16使用双标杆 addcmul fp16使用bf16的标准 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5181	1 个月前
codegen	fix docs error Co-authored-by: bellatan<tanmei2@huawei.com> # message auto-generated for no-merge-commit merge: !5466 merge 0706_docs_v26.1.0 into 26.1.0 fix docs error Created-by: bellatan Commit-by: bellatan Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> https://gitcode.com/Ascend/op-plugin/issues/266 - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 \| api名字 \| 文档 \| 问题 \| 修改处 \| \|---\|---\|---\|---\| \| `at_npu::native::get_npu_format` \| `（beta）at_npu-native-get_npu_format.md` \| 定义文件路径使用了 Windows 反斜杠；功能说明中“获取NPU tensor格式信息，返回值类型int64_t”表述不顺，需明确为获取格式信息且返回值为 int64_t 类型。 \| `torch_npu\csrc\core\npu\NPUFormat.h` 改为 `torch_npu/csrc/core/npu/NPUFormat.h`；功能说明改为“获取NPU tensor的格式信息，返回值为int64_t类型。” \| \| `torch_npu.contrib.function.fuse_add_softmax_dropout` \| `（beta）torch_npu-contrib-function-fuse_add_softmax_dropout.md` \| 示例代码把第五个入参写成 `alpha`，与接口含义不匹配；应改为 `attn_head_size`，并补齐 `p=0.1`。 \| 示例中 `alpha = 0.125` 改为 `attn_head_size = 64`；调用参数 `alpha` 改为 `attn_head_size, p=0.1`。 \| \| `torch_npu.contrib.function.npu_bbox_coder_decode_xywh2xyxy` \| `（beta）torch_npu-contrib-function-npu_bbox_coder_decode_xywh2xyxy.md` \| 功能描述用语不准确；示例中 `torch.randint(...).npu()` 生成整型 bboxes，不符合 bbox 坐标应使用浮点张量的示例需求。 \| 功能说明改为“将格式从xywh转换为xyxy”；`torch.randint(0, max_shape, size=(A, 4)).npu()` 改为 `(torch.rand(A, 4) * max_shape).npu()`。 \| \| `torch_npu.contrib.function.npu_ciou` \| `（beta）torch_npu-contrib-function-npu_ciou.md`<br>`torch_npu-contrib_list.md` \| 功能说明存在英文混杂和语义不顺的问题，如 `penalty term`、`propose CIoU`，需要改成中文描述。 \| `penalty term` 改为“惩罚项”；`propose CIoU` 改为“提出CIoU”。 \| \| `torch_npu.contrib.module.ROIAlign` \| `（beta）torch_npu-contrib-module-ROIAlign.md` \| 示例代码对 `roi` 设置 `requires_grad = True`，但 ROIAlign 对 rois 入参不支持求导，示例会误导用户。 \| 删除示例中的 `roi.requires_grad = True`。 \| \| `torch_npu.npu.obfuscation_calculate` \| `（beta）torch_npu-npu-obfuscation_calculate.md`<br>`_op_plugin_docs.py` \| `x` 参数 shape 写成 `( , , ... , hiddenSize)`，格式异常；应表达为 `(, ..., hiddenSize)`。 \| `( , , ... , hiddenSize)` 改为 `(, ..., hiddenSize)`。 \| \| `torch_npu.npu.set_compile_mode` \| `（beta）torch_npu-npu-set_compile_mode.md`<br>`torch_npu-npu_list.md` \| 文档把 `jit_compile` 解释为“二进制/非二进制模式”，语义不直观；应按开启或关闭 JIT 编译描述，并同步产品默认值说明。 \| “设置是否开启二进制编译”改为“设置是否开启JIT编译”；`True/False` 说明改为开启或关闭 JIT 编译。 \| \| `torch_npu.npu_incre_flash_attention` \| `torch_npu-npu_incre_flash_attention.md`<br>`_op_plugin_docs.py` \| page attention 性能说明中“带来性能上的下降”表述笼统；示例把环境变量写成裸赋值，放在 Python 代码块中不可执行。 \| “会带来性能上的下降”改为“可能会增加延迟”；`TORCHDYNAMO_VERBOSE=1` 和 `TORCH_LOGS="+dynamo"` 改为 `os.environ[...]` 写法。 \| \| `torch_npu.npu_swiglu_quant` \| `torch_npu-npu_swiglu_quant.md` \| `group_index` 说明只写成“每个 group 分组的 tokens 数”，没有体现其含义会随 `group_list_type` 模式变化。 \| `group_index` 说明改为“用于定义分组边界或每组tokens数（具体取决于group_list_type模式）”。 \| \| `torch.npu.set_device_limit` \| `torch_npu-set_device_limit.md` \| `device` 参数类型写成 `Device`，与接口原型和实际用法不一致，应为 `int` 卡号。 \| `device` 参数类型由 `Device` 改为 `int`。 \| \| `torch.npu.set_stream_limit` \| `torch_npu-set_stream_limit.md` \| 功能说明误写为依赖 `torch.npu.set_device_limit`；参数默认值说明使用“分核”不准确；示例缺少先设置 device 的调用。 \| `torch.npu.set_device_limit` 改为 `torch.npu.set_device`；“默认为-1不设置分核”改为“默认为-1不设置控核”；示例增加 `torch.npu.set_device(0)`。 \| \| `torch_npu.npu_ciou` \| `（beta）torch_npu-npu_ciou.md`<br>`torch_npu_list.md`<br>`_op_plugin_docs.py` \| 功能说明英文混杂；函数原型使用 `self, gtboxes` 与参数说明 `boxes1, boxes2` 不一致；`is_cross` 描述和返回值说明也沿用了错误对象或“mask操作”表述。 \| `self, gtboxes` 改为 `boxes1, boxes2`；`box1和box2` 改为 `boxes1和boxes2`；“mask操作的结果”改为“CIoU计算结果”。 \| \| `torch_npu.npu_dropout_with_add_softmax` \| `（beta）torch_npu-npu_dropout_with_add_softmax.md`<br>`_op_plugin_docs.py` \| 产品支持表残留 `<term>` 标签；公式说明把输入写成 `x`，与函数原型中的 `self` 不一致；mask 说明把丢弃概率写成 `p`，应对应参数 `prob`。 \| 删除产品支持表中的 `<term>` 标签；`x` 改为 `self`；`丢弃概率p` 改为 `丢弃概率prob`。 \| 其中 `torch.npu.set_stream_limit`的修改主要参考CANN资料手册说明，https://www.hiascend.com/document/detail/zh/CANNCommunityEdition/910beta3/API/runtimeapi/aclcppdevg_03_1947.html # 【资料变更】修改内容如【修改方案】展示 # 【接口变更】不涉及 # 【功能验证】 ci pass # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5466	16 天前
docs	modify document Co-authored-by: m0_73734473<liudingwen1@h-partners.com> # message auto-generated for no-merge-commit merge: !5617 merge 26.1.0 into 26.1.0 modify document Created-by: m0_73734473 Commit-by: m0_73734473 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/pytorch/issues/3320 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 aidd问题扫描修改 # 【资料变更】涉及 # 【接口变更】不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5617	15 小时前
examples	modify document Co-authored-by: m0_68000167<wangzhiyuan86@h-partners.com> # message auto-generated for no-merge-commit merge: !5576 merge 26.1.0 into 26.1.0 modify document Created-by: m0_68000167 Commit-by: m0_68000167 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/op-plugin/issues/320 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】文件修改 # 【资料变更】涉及 # 【接口变更】不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5576	4 天前
op_plugin	[fix] 兼容性修改 Co-authored-by: l00881990<lixinyu68@huawei.com> # message auto-generated for no-merge-commit merge: !5516 merge cherry-pick-mr-5512-1784623751094-auto into 26.1.0 [fix] 兼容性修改 Created-by: l1919_snow Commit-by: l00881990 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 A5 和A2 区分开，A5 走新分支，A2 a3 原分支 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 “不涉及” # 【功能验证】 A5 不受影响 image.png A3走index_add # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5516	10 天前
pre-commit	[Fix] Fix error of code check Co-authored-by: huangjingwei<huangjingwei4@huawei.com> # message auto-generated for no-merge-commit merge: !5026 merge code_check into master [Fix] Fix error of code check Created-by: huangjingwei Commit-by: huangjingwei Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [x] 重构优化 - [ ] 资料更新 # 【修改方案】支持部分检查工具，并完成全仓整改。 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7403085/3269f826-aa4b-4bc7-b7f6-7733f1034fb3/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5026	2 个月前
test	delete quant_all_reduce & quant_reduce_scatter Co-authored-by: yifux<xiongyifu1@huawei.com> # message auto-generated for no-merge-commit merge: !5477 merge pr_2610 into 26.1.0 delete quant_all_reduce & quant_reduce_scatter Created-by: xiongyifu Commit-by: yifux Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 https://gitcode.com/Ascend/op-plugin/issues/285 - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 master分支，26.1.0分支删除quant_all_reduce和quant_reduce_scatter算子 # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】不涉及，仅删除算子不影响工程编译 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5477	14 天前
torchnpugen	Fix memory exceptions in BEVFusion Co-authored-by: zzhongmin<zhongmin23@huawei.com> # message auto-generated for no-merge-commit merge: !5379 merge 26.1.0_fixcat into 26.1.0 Fix memory exceptions in BEVFusion Created-by: zzhongmin Commit-by: zzhongmin Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [x] 问题单https://gitcode.com/Ascend/pytorch/issues/2597 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】回退https://gitcode.com/Ascend/op-plugin/pull/5048对cat的改动 1. 恢复 cat 非出参版本的内核实现 - op_plugin/ops/aclops/CatKernelNpu.cpp:恢复 acl_op::cat(const at::ITensorListRef&, int64_t); - op_plugin/ops/opapi/CatKernelNpuOpApi.cpp:恢复 op_api::cat(const at::ITensorListRef&, int64_t)。 2. 恢复 cat 的接口声明生成修改 torchnpugen/gen.py 的 should_generate_op_interface,对 base name 为 cat 的算子放行(返回 True),使其照常生成 OpInterface/OpApiInterface/AclOpsInterface 声明,与 gen_backend_stubs.py （torch_npu）中对 cat 的特判保持一致: # 【资料变更】不涉及 # 【接口变更】不涉及 # 【功能验证】 BEVFusion网络显存正常 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5379	27 天前
.clang-format	Add static code check and support pre-commit Co-authored-by: root<Jingwei Huang> # message auto-generated for no-merge-commit merge: !4793 merge code-check into master Add static code check and support pre-commit Created-by: huangjingwei Commit-by: root Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [x] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容）对接PUD静态检查工具和规则。支持pre-commit。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图直接运行pre-commit命令： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/363e7457-d1fe-4deb-a751-345ee1facd85/image.png 'image.png') git cherry-pick 或 git commit 自动运行代码静态检查： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/add96b31-7766-4d1d-9f6a-61fd10ca51eb/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!4793	3 个月前
.gitignore	init submodule	3 年前
.pre-commit-config.yaml	[Fix] Enable trailing whitespace check, Fix trailing-whitespace errors detected by pre-commit Co-authored-by: zhoupeng<zp2165064878@163.com> # message auto-generated for no-merge-commit merge: !5043 merge op_code_check into master [Fix] Enable trailing whitespace check, Fix trailing-whitespace errors detected by pre-commit Created-by: thickhair Commit-by: zhoupeng Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 https://gitcode.com/Ascend/op-plugin/issues/129 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 修改pre-commit-config.yaml，开启trailing-whitespace检查，使用pre-commit进行检查和修复 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7403085/549c5212-645f-4d66-82e2-88926bc8515e/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5043	1 个月前
CONTRIBUTING.md	Add static code check and support pre-commit Co-authored-by: root<Jingwei Huang> # message auto-generated for no-merge-commit merge: !4793 merge code-check into master Add static code check and support pre-commit Created-by: huangjingwei Commit-by: root Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [ ] issue/工单 - [x] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容）对接PUD静态检查工具和规则。支持pre-commit。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图直接运行pre-commit命令： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/363e7457-d1fe-4deb-a751-345ee1facd85/image.png 'image.png') git cherry-pick 或 git commit 自动运行代码静态检查： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/add96b31-7766-4d1d-9f6a-61fd10ca51eb/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!4793	3 个月前
LICENSE	update LICENSE. Signed-off-by: weili10 <liwei386@huawei.com>	2 年前
OWNERS	Update URL (gitcode) Co-authored-by: dilililiwhy<why.wuhuanyu@huawei.com> # message auto-generated for no-merge-commit merge: merge update_url into master Update URL (gitcode) Created-by: dilililiwhy Commit-by: dilililiwhy Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> What type of PR is this? > Uncomment only one `/kind <>` line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature What does this PR do / why do we need it: Special notes for your reviewers: See merge request: Ascend/op-plugin!3165	10 个月前
README.md	modify document Co-authored-by: m0_68000167<wangzhiyuan86@h-partners.com> # message auto-generated for no-merge-commit merge: !5536 merge 26.1.0 into 26.1.0 modify document Created-by: m0_68000167 Commit-by: m0_68000167 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">如有社区issue，请关联issue链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部issue等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/pytorch/issues/1880 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】文件修改 # 【资料变更】涉及 # 【接口变更】不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!5536	8 天前
Third_Party_Open_Source_Software_Notice	!2388 [master] Update the referenced PyTorch version to v2.3.1. Merge pull request !2388 from shaoyf/master_notice	1 年前
gencode.sh	Generate ATB public C++ interface wrappers from dispatcher schema registrations. Co-authored-by: wang_ziqi<wangziqi4@huawei.com> # message auto-generated for no-merge-commit merge: !4825 merge atb-cpp-api-interface into master Generate ATB public C++ interface wrappers from dispatcher schema registrations. Created-by: wang-ziqi-code Commit-by: wang_ziqi Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：2026-04-25 --> # 【合入来源】> <font color="red">如有社区 issue，请关联 issue 链接</font>\ > <font color="red">请勿携带内部流程信息（需求链接、问题单、内部 issue 等）</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 https://gitcode.com/Ascend/op-plugin/issues/69 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】本次补充 ATB C++ 公共接口暴露机制，并同步收敛配置和生成器口径。当前 ATB 算子原本主要通过 `TORCH_LIBRARY_FRAGMENT(atb, m)` / `m.impl(...)` 注册到 dispatcher，再由 Python 侧通过 `torch.ops.atb` 使用；本次在此基础上新增 `at_npu::native::atb` 命名空间下的 C++ public interface，用于 `torch_npu` 同构环境中的 C++ 调用方直接访问。整体链路如下： `text ::atb::implementation -> TORCH_LIBRARY_FRAGMENT(atb, m).def(...) / m.impl(...) -> torch.ops.atb (Python dispatcher entry) #python层接口 -> atb_ops.yaml + gen_atb_ops.py -> at_npu::native::atb::* (public C++ interface) #cpp层接口` 具体修改如下： 1. 新增并收敛 ATB C++ 接口生成器 `torchnpugen/gen_atb_ops.py` - 以 `::atb::` 实现函数签名作为 public C++ interface 的真值来源； - 以 `m.def(...)` / `m.impl(...)` 作为注册约束校验； - 读取 `op_plugin/config/atb_ops.yaml` 中的 `cpp_name -> impl_name` 映射，生成： - `op_plugin/include/AtbOpsInterface.h` - `op_plugin/ops/atb/AtbOpsInterface.cpp` - 生成器会校验： - 对应 `::atb::` 实现签名可从 `op_plugin/ops/atb/.cpp` 中唯一解析。 2. 新增配置文件 `op_plugin/config/atb_ops.yaml`* - 维护 public C++ interface 与 `::atb::` 实现之间的映射； - 当前共覆盖 34 个 ATB 接口； - 配置写法为： `yaml - cpp_name: "at_npu::native::atb::_npu_reshape_and_cache" impl_name: "atb::_npu_reshape_and_cache"` - 生成器内部会解析并校验 `cpp_name`： - 必须以 `at_npu::native::atb::` 开头； - 必须能提取出合法的末尾函数名； 3. 新增模板文件 `torchnpugen/templates/AtbOpsInterface.h` 和 `AtbOpsInterface.cpp`* - 通过 `string.Template` 渲染 public declaration、forward declaration 和 wrapper definition； - 生成的 wrapper 统一位于 `namespace at_npu::native::atb` 下； - wrapper 最终按参数原样转发到对应 `::atb::` 实现。 4. 新增统一 include 入口 `op_plugin/include/atb_ops.h`* - 作为 C++ 使用方的统一头文件入口； - 当前仅包含 `AtbOpsInterface.h`。 5. 集成到代码生成流程 - 在现有代码生成链路中加入 `gen_atb_ops.py`； - 保证构建过程中自动生成最新接口文件。 ### 生成产物示例以 `_npu_group_topk` 为例，生成结果如下： AtbOpsInterface.h `cpp namespace at_npu { namespace native { namespace atb { TORCH_NPU_API void _npu_group_topk(const at::Tensor &self, int64_t k, int64_t group_num, int64_t n); } // namespace atb } // namespace native } // namespace at_npu` AtbOpsInterface.cpp `cpp namespace atb { void _npu_group_topk(const at::Tensor &self, int64_t k, int64_t group_num, int64_t n); } // namespace atb namespace at_npu { namespace native { namespace atb { void _npu_group_topk(const at::Tensor &self, int64_t k, int64_t group_num, int64_t n) { ::atb::_npu_group_topk(self, k, group_num, n); } } // namespace atb } // namespace native } // namespace at_npu` # 【资料变更】不涉及 # 【接口变更】涉及 C++ public interface 新增。新增 `AtbOpsInterface.h`，在 `at_npu::native::atb` 命名空间下暴露 34 个 ATB public C++ API；接口签名与对应 `::atb::` 实现保持一致。C++ 调用方通过： `cpp #include "atb_ops.h"` 即可访问对应接口，链接目标为 `libop_plugin_atb.so`。 - 在 `torch_npu` 同构运行环境中，`at_npu::native::atb::` 作为额外 C++ ABI 可用； - 接口签名与 `::atb::` 实现保持一致； - public 符号边界位于 `libop_plugin_atb.so`。 # 【功能验证】 1. 生成回归* - 执行 `python -m torchnpugen.gen_atb_ops --config op_plugin/config/atb_ops.yaml --header op_plugin/include/AtbOpsInterface.h --source op_plugin/ops/atb/AtbOpsInterface.cpp --atb-src-dir op_plugin/ops/atb` - 连续执行两次，确认生成结果稳定、无无关 diff。 2. 源码契约检查 - 执行 `python test/check_atb_cpp_api_contract.py --check-source` - 校验： - `cpp_name -> impl_name -> m.impl -> m.def` 映射成立； - wrapper 声明与 `::atb::` 实现签名一致； - wrapper 参数透传顺序一致。 3. 配置校验* - `atb_ops.yaml` 中 `cpp_name` 采用全限定写法； - 非法 `cpp_name`（缺前缀、缺函数名）会在生成器配置解析阶段直接报错。 4. 编译链接验证 - `op-plugin` 全量编译通过； - `at_npu::native::atb::*` public 符号由 `libop_plugin_atb.so` 承载。 # 【CheckList】 > PR 提交人对以下 CheckList 自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR 标题正确使用类型标签，如：feat、fix、refactor、docs、test 等 - [x] PR 持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!4825	2 个月前

OpPlugin

简介

本项目开发了TorchNPU算子插件，为使用PyTorch框架的开发者提供便捷的NPU算子库调用能力。 OpPlugin算子插件的编译、使用依赖昇腾TorchNPU。因此，在编译OpPlugin之前，需要了解、安装昇腾PyTorch。使用手册可参考昇腾社区TorchNPU。

目录结构

关键目录如下:

├─docs                             # 文档目录
├─ci                               # 自动化构建与测试脚本目录
├─op_plugin                        # 项目核心目录
│  ├─config                        # 配置管理目录
│  ├─ops                           # 算子实现目录
│  ├─python                        # python绑定目录
├─codegen                          # 代码生成目录
├─examples                         # 示例目录
└─test                             # 测试目录

版本配套表

OpPlugin仓旨在为TorchNPU提供运行所需要的算子适配文件，两个仓的对应关系如下：

OpPlugin分支	对应TorchNPU版本
master	主线版本，如v2.7.1等
26.1.0	26.1.0版本，如v2.7.1-26.1.0等
26.0.0	26.0.0版本，如v2.7.1-26.0.0等
7.3.0	7.3.0版本，如v2.7.1-7.3.0等
7.2.0	7.2.0版本，如v2.7.1-7.2.0等
7.1.0	7.1.0版本，如v2.1.0-7.1.0等
7.0.0	7.0.0版本，如v2.1.0-7.0.0等
6.0.0	6.0.0版本，如v2.1.0-6.0.0等
6.0.rc3	6.0.rc3版本，如v2.1.0-6.0.rc3等
6.0.rc2	6.0.rc2版本，如v2.1.0-6.0.rc2等
6.0.rc1	6.0.rc1版本，如v2.1.0-6.0.rc1等
5.0.0	5.0.0版本，如v2.1.0-5.0.0等
5.0.rc3	5.0.rc3版本，如v2.1.0-5.0.rc3等

安装OpPlugin

支持通过源码编译的方式安装OpPlugin。具体操作，请参考安装OpPlugin。

快速入门

提供了一个通过OpPlugin插件实现PyTorch调用Ascend C自定义算子的完整开发指南，涵盖了从环境配置、算子注册、适配实现到测试验证提供了全流程说明。具体操作，请参考调用样例。

API参考

基于PyTorch2.10.0/2.9.0/2.8.0/2.7.1版本，提供TorchNPU自定义API的功能说明、函数原型、参数说明与调用示例等。具体信息，请参考自定义API。

生命周期

OpPlugin仓依赖TorchNPU运行，生命周期请参考TorchNPU中的PyTorch版本维护策略。

贡献指导

介绍如何向OpPlugin仓库贡献代码，具体请参见贡献指南。

联系我们

如果有任何疑问或建议，请提交GitCode Issues，我们会尽快回复。感谢您的支持。

安全声明

主要描述了OpPlugin的安全加固信息、公网地址信息及通信矩阵等内容。具体介绍，请参考OpPlugin安全声明。

免责声明

致OpPlugin插件使用者

本插件仅供调试和开发使用，使用者需自行承担使用风险，并理解以下内容：
- 数据处理及删除：用户在使用本插件过程中产生的数据属于用户责任范畴。建议用户在使用完毕后及时删除相关数据，以防信息泄露。
- 数据保密与传播：使用者了解并同意不得将通过本插件产生的数据随意外发或传播。对于由此产生的信息泄露、数据泄露或其他不良后果，本插件及其开发者概不负责。
- 用户输入安全性：用户需自行保证输入的命令行的安全性，并承担因输入不当而导致的任何安全风险或损失。对于输入命令行不当所导致的问题，本插件及其开发者概不负责。
免责声明范围：本免责声明适用于所有使用本插件的个人或实体。使用本插件即表示您同意并接受本声明的内容，并愿意承担因使用该功能而产生的风险和责任，如有异议请停止使用本插件。
在使用本工具之前，请谨慎阅读并理解以上免责声明的内容。对于使用本插件所产生的任何问题或疑问，请及时联系开发者。

License

OpPlugin的使用许可证，详见LICENSE。

致谢

感谢来自社区的每一个PR，欢迎贡献Ascend Extension for TensorPipe插件！