已关闭
[Bug-Report|缺陷反馈]: SwigluGroupGrad Torch Extension 目录不符合算子就近维护规范 #4433
sinobr创建于  7月30日关闭于  7月31日
sinobr
sinobr
7月30日 创建

Describe the current behavior / 问题描述 (Mandatory / 必填)

SwigluGroupGrad 算子已完成上库,但 Torch Extension 适配代码位于仓库公共目录 torch_extension/cann_ops_nn/,未放置在算子自身目录下,不符合算子代码就近维护的目录规范,也不利于算子代码、Torch 注册、TorchAir 转换和 ACLNN 适配的统一维护。

相关修复 PR:8096

Environment / 环境信息 (Mandatory / 必填)

  • 问题类型:源码目录及 Torch Extension 打包结构,不涉及特定昇腾硬件或运行时版本

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

  1. 查看 torch_extension/cann_ops_nn/ops/activation/swiglu_group_backward/
  2. 查看 torch_extension/cann_ops_nn/csrc/activation/swiglu_group_backward.cpp
  3. 对比已有 operator-local Torch Extension 算子 quant/mx_to_block_mx_quant/torch_extension/
  4. 可以看到 SwigluGroupGrad 的 Torch Extension 文件位于公共目录,而不是 quant/swiglu_group_grad/ 算子目录。

Describe the expected behavior / 预期结果 (Mandatory / 必填)

将 SwigluGroupGrad 的 Torch Python 注册、TorchAir 图转换、C++ ACLNN 适配和 Torch API 文档迁移至 quant/swiglu_group_grad/ 下;构建系统继续将 operator-local 文件正确收集到完整 wheel 和单算子 wheel 中,并保持 torch.ops.cann_ops_nn.swiglu_group_backward 对外接口及计算逻辑不变。

问题路径:

  • torch_extension/cann_ops_nn/ops/activation/swiglu_group_backward/
  • torch_extension/cann_ops_nn/csrc/activation/swiglu_group_backward.cpp

预期路径:

  • quant/swiglu_group_grad/torch_extension/
  • quant/swiglu_group_grad/docs/torchapi_swiglu_group_backward.md

修复和验证记录见 PR:cann/ops-nn!8096

likedislike
sinobrsinobr
7月30日 添加了label:bug-report
sinobrsinobr
7月30日 关联了pull request:修复 SwigluGroupGrad 算子 Torch Extension 目录结构
sinobrsinobr
7月30日 修改了issue 的描述
sinobrsinobr
7月30日 修改了issue 的描述
yuning_chenyuning_chen成员
7月30日 将 qq_51545127 设为负责人
CANN-robotCANN-robot成员
7月31日 关闭了 issue
CANN-robotCANN-robot成员
7月31日 添加了label:resolved