已合并
[CANNBot]:新增ActULQClampMinGrad算子适配 Ascend950 #8762
Nerddddddddddd创建于 8月17日
[CANNBot]:新增ActULQClampMinGrad算子适配 Ascend950 #8762
已合并
Nerddddddddddd创建于 8月17日
Nerddddddddddd
Nerddddddddddd成员
8月17日

描述

  • 算子功能:用于ULQ(Ultra-Low-bit Quantization,超低比特量化)量化感知训练中,计算clamp下界截断的反向梯度。该算子对输入张量执行全轴归约求和操作,输出一个标量梯度值。
    • 计算公式:

      • y_grady\_grad:来自后续层的梯度张量
      • clamp_min_maskclamp\_min\_mask:clamp下界掩码(布尔型或浮点型,取值为0或1)
      • x_clamped_lossx\_clamped\_loss:经过clamp后的损失值张量
      • clamp_min_gradclamp\_min\_grad:输出的标量梯度(0维张量)

关联的Issue

https://gitcode.com/cann/ops-nn/issues/4366

测试

文档更新

类型标签

AI/Agent生成声明

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 Nerddddddddddd 的贡献)
NerdddddddddddNerddddddddddd成员
8月17日 创建了 pull request,commit 7b84601a
atomgit-bot
atomgit-bot
8月17日 评论:

🤖 正在生成合并请求摘要,请稍候…

likedislike
不准确?
atomgit-bot
atomgit-bot
8月17日 评论:

🤖 AI 代码检视正在进行中,请稍候…

likedislike
不准确?
CANN-robotCANN-robot成员
8月17日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
8月17日 评论:

CLA Signature Pass

Coder_Nerd, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了129条消息 查看更多
CANN-robotCANN-robot成员
8月18日 添加了label:lgtm
tang-lei01成员
8月18日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
8月18日 添加了label:approved
CANN-robotCANN-robot成员
8月18日 合入了pull request
CANN-robot
CANN-robot成员
8月18日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike