已合并
新增ThresholdBackward实验算子(relu_grad_v2) #4479
cwzhang创建于 4月30日
新增ThresholdBackward实验算子(relu_grad_v2) #4479
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 cwzhang 的贡献)CANN-robot
4月30日 评论:
4月30日 评论:
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| experimental/activation | ✅ 周奇龙, 查建青 (2/2) | ✅ 周奇龙, 查建青 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
cwzhang, thanks for your pull request. All authors of the commits have signed the CLA. 👍


4月30日 添加了label:cann-cla/no
4月30日 将crystalhu,yangyang016,fanqirui,zhou-qilong,zhajianqing123,chenqi317,liubo75,tangweiwei2,liujie12345678,Chen_HaoWen,wangyongguang,zhang-wu设为评审人
4月30日 将crystalhu,yangyang016,fanqirui,zhou-qilong,zhajianqing123,chenqi317,liubo75,tangweiwei2,liujie12345678,Chen_HaoWen,wangyongguang,zhang-wu设为评审人
此处折叠了91条消息 查看更多
zhajianqing123
5月13日 评论:
5月13日 评论:
/lgtm
/approve


5月13日 添加了label:lgtm
CANN-robot
5月13日 评论:
5月13日 评论:
5月13日 关闭了关联的issue
5月13日 合入了pull request
描述
新增 ThresholdBackward 实验性算子,位于
experimental/activation/relu_grad_v2,用于计算 Relu 激活函数的反向传播,公式为:当输入 self > threshold 时输出 gradOutput,否则输出 0。TBE转AscendC算子迁移项目,支持该算子后续开发与演进。
experimental/activation/relu_grad_v2aclnnThresholdBackward关联的Issue
[Requirement|需求建议]: 新增ThresholdBackward实验算子(relu_grad)(关联PR用于追踪)
测试
UT 测试 17 / 17 通过
测试命令:
bash build.sh --experimental --ops=relu_grad_v2 -u --opapi -j8 -O2测试覆盖以下场景:
empty_tensornot_contiguousnullptrthreshold != 0ATK 精度测试 2100 / 2100 通过
文档更新
docs/aclnnThresholdBackward.mdREADME.mdexamples/test_aclnn_relu_grad_v2.cppexamples/run.sh类型标签